Name		Name	Last commit message	Last commit date
parent directory ..
LICENSE		LICENSE
README.md		README.md
pizza.jpg		pizza.jpg
volo.py		volo.py
volo_labels.py		volo_labels.py

README.md

VOLO

Input

Shape : (1,3,224,224)

Output

class_count=3
+ idx=0
  category=963[pizza, pizza pie ]
  prob=12.82837963104248
+ idx=1
  category=923[plate ]
  prob=3.2662553787231445
+ idx=2
  category=572[goblet ]
  prob=2.8674607276916504
Script finished successfully.

Usage

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 volo.py

If you want to specify the input image, put the image path after the --input option.

$ python3 volo.py --input IMAGE_PATH

By adding the --video option, you can input the video.
If you pass 0 as an argument to VIDEO_PATH, you can use the webcam input instead of the video file.