How to run projects/CenterNet2/demo.py

[SHIsue]

1. download zip and unzip
2. makesure your torch>=1.6
3. open INSTALL.md and install Pre-Built Detectron2. For example, if your torch==1.6 & cuda==10.1, then run "python -m pip install detectron2 -f
   https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.6/index.html"
4. cd CenterNet2-master
5. python setup.py develop
6. vim projects/CenterNet2/demo.py
7. Comment out all cv2.imshow & cv2.nameWindow & cv2.waitkey() & cv2.destroyAllWindows() and save demo.py
8. back to CenterNet2-master/ and run "CUDA_VISIBLE_DEVICES=5 python projects/CenterNet2/demo.py --config-file projects/CenterNet2/configs/CenterNet2_DLA-BiFPN-P3_24x.yaml --input datasets/imgs/ --opts MODEL.WEIGHTS ../models/CenterNet2_DLA-BiFPN-P3_24x.pth" (make sure you have imgs in datasets/imgs/ and download the model in models/ , Just An Example!!!!!)

[shengyuan-tang]

How do you find the 'CenterNet2_DLA-BiFPN-P3_24x.pth', i can't find this weights.

[SHIsue]

https://github.com/xingyizhou/CenterNet2/blob/master/projects/CenterNet2/centernet2_docs/MODEL_ZOO.md