Skip to content

Latest commit

 

History

History
51 lines (33 loc) · 1.24 KB

README.md

File metadata and controls

51 lines (33 loc) · 1.24 KB
Raw

CDNet

Input

Input

(Image from https://github.com/zhangzhengde0225/CDNet/tree/master/example/images)

Output

Output

If you input a video or multiple images, in addition to the bounding box, the result of the vector crossing method is drawn.

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 cdnet.py --input example --savepath output

If you want to specify the input image, put the image path after the --input option.
You can use --savepath option to change the name of the output file to save.

$ python3 cdnet.py --input IMAGE_PATH --savepath SAVE_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.

$ python3 cdnet.py --video VIDEO_PATH

Reference

Framework

Pytorch

Model Format

ONNX opset=11

Netron

cdnet.onnx.prototxt