Hi
I’m a beginner using PyTorch models
I need to know the name or id image prediction and store the results inside CSV file (id or name image and predictions)
def visualize_model(model, num_images=10):
was_training = model.training

model.eval()



images_so_far = 0

fig = plt.figure()

with torch.no_grad():

    for i, (inputs, labels) in enumerate(dataloaders['val']):

        inputs = inputs.to(device)

        labels = labels.to(device)

        outputs = model(inputs)

        _, preds = torch.max(outputs, 1)

    

        for j in range(inputs.size()[0]):

       

            images_so_far += 1

            ax = plt.subplot(num_images//2, 2, images_so_far)

            ax.axis('off')

            ax.set_title('predicted: {}'.format(class_names[preds[j]]))

            imshow(inputs.cpu().data[j])

            if images_so_far == num_images:

                model.train(mode=was_training)

                return

    model.train(mode=was_training)

the code prints the image and predicts, is there a way to retrieve the name or id image?
sorry I am a beginner in vision