Train the model by running: cmd :: 'python train.py' The model is stored in 'snapshot.pth' file. We can now test individual images(test.jpg in this example) using test.py script by running: cmd:: 'python test.py test.jpg'