GitHub - ahmedshoaib/fmd

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
fmd/image		fmd/image
readme.txt		readme.txt
snapshot.pth		snapshot.pth
snapshot25.pth		snapshot25.pth
test.jpg		test.jpg
test.py		test.py
test1.jpg		test1.jpg
test2.jpg		test2.jpg
test3.jpg		test3.jpg
test4.jpg		test4.jpg
train.py		train.py

Repository files navigation

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'