(Image from http://farm4.staticflickr.com/3693/9472793441_b7822c00de_z.jpg)
This model requires additional module.
pip3 install transformers
pip3 install nltk
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 glip.py
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 glip.py --input IMAGE_PATH --savepath SAVE_IMAGE_PATH
If you want to specify the caption for detection, put the caption after the --caption
option.
$ python3 glip.py --caption "bobble heads on top of the shelf"
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 glip.py --video VIDEO_PATH
Pytorch
ONNX opset=12
swin_tiny_bert.onnx.prototxt
swin_tiny_backbone.onnx.prototxt
swin_tiny_rpn.onnx.prototxt