How to output the corresponding original picture after classification prediction?

xiao · December 12, 2018, 2:45pm

Recently, I used my trained model to predict and classify a large number of new images, but I can only predict the type of each image. I can't find the original image of this image. My purpose is to sorted and saved each picture . Since my data was preprocessed by cropped and flipped, I don't know how to restore them.
I don't know how to do it, if you konw how to restore the pictures, please help me.
Any suggestions would be really helpful. Thanks in advance.

Amrit_Das · December 12, 2018, 3:06pm

carrying Inference on a trained image, jsut predicts the class of the image. For example in my case,

github.com

amrit-das/Custom-Model-Training-PyTorch/blob/master/predict.py

import torch
import torch.nn as nn
from torchvision.models import resnet18
from torchvision.transforms import transforms
import matplotlib.pyplot as plt
import numpy as np
from torch.autograd import Variable
import torch.functional as F
from PIL import Image
import os
import sys
import argparse

parser = argparse.ArgumentParser(description = 'To Predict from a trained model')
parser.add_argument('-i','--image', dest = 'image_name', required = True, help='Path to the image file')
parser.add_argument('-m','--model', dest = 'model_name', required = True, help='Path to the model')
parser.add_argument('-n','--num_class',dest = 'num_classes', required = True, help='Number of training classes')
args = parser.parse_args()

path_to_model = "./models/"+args.model_name

This file has been truncated. show original

I read one image at a time. You can easily automate this by writing another script that reads images from a folder one by one, predicts them and then based on the predicted class stores them in a new directory.

I hope I got your question because I was a bit confused first.

xiao · December 13, 2018, 10:57am

Thank you very much, I am sorry, my expression is not clear. Luckily, you got my question, In fact, I have been able to read and predict one image at a time, you are right, I need write another script that reads images from a folder one by one, Now I am trying to do this, thank you.

Amrit_Das · December 15, 2018, 6:11am

@xiao I would request you to please mark this discussion as solved so that it might be helpful for the further coming members to this post!