Prerequisites:
PyTorch must be installed. To install PyTorch, do pip install torch
and pip install torchvision
in the terminal.
This image recognition program sees images & tells what it sees in the image.
To run the script with the given 20 images in the folder SMALL DATASET, use the following command:
python Resnet_image_recognition.py 7.jpeg
This command will run the script with the image 7.jpeg. You can change this number from 1.jpeg to 20.jpeg
To make your program recognize images given by you, add some image, you want this program to recognize, in folder SMALL DATASET. Then use the following command to run the script with your image:
python Resnet_image_recognition.py my_pic.png
This command will run the script with your image my_pic.png