Medical_Image_Deep_Learning_Assignment_1

A simple neural network written in Pytorch to locate a cellphone an image input

Sample Commands

To test a specific image

python3 src/test.py ./test/121.jpg

To test all of the images in ./test

python3 src/test.py