Breaking News: Grepper is joining You.com. Read the official announcement!

set threshold resnet18 pytorch

Merwanski answered on January 19, 2022 Popularity 1/10 Helpfulness 8/10

answer set threshold resnet18 pytorch

set threshold resnet18 pytorch

Comment

from PIL import Image
import torch
from torchvision import transforms
input_image = Image.open("img/danbooru_resnet1.png") # load an image of your choice
preprocess = transforms.Compose([
    transforms.Resize(360),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.7137, 0.6628, 0.6519], std=[0.2970, 0.3017, 0.2979]),
])
input_tensor = preprocess(input_image)
input_batch = input_tensor.unsqueeze(0) # create a mini-batch as expected by the model

if torch.cuda.is_available():
    input_batch = input_batch.to('cuda')
    model.to('cuda')

with torch.no_grad():
    output = model(input_batch)

# The output has unnormalized scores. To get probabilities, you can run a sigmoid on it.
probs = torch.sigmoid(output[0]) # Tensor of shape 6000, with confidence scores over Danbooru's top 6000 tags

# Second part of the code 
# to set a threshold value and to use it 
tmp = probs[probs > thresh]
inds = probs.argsort(descending=True)

Popularity 1/10 Helpfulness 8/10 Language python

Source: github.com

Tags: python pytorch set threshold

Link to this answer
Share Copy Link

Contributed on Jan 19 2022

Merwanski

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

set threshold resnet18 pytorch

Contents

More Related Answers

set threshold resnet18 pytorch

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.