Input file type must be .jpg image.
Input file type must be .mp4 video.
class_count=3
+ idx=0
category=409[analog clock ]
prob=0.8660851716995239
+ idx=1
category=892[wall clock ]
prob=0.13281846046447754
+ idx=2
category=826[stopwatch, stop watch ]
prob=0.0008276691660284996
Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.
For the sample image,
$ python3 alexnet.py --input [your sample.jpg path]
For the sample video,
$ python3 alexnet.py --video [your sample.mp4 path]
ONNX opset = 11
Pytorch 1.7.1