Faster R-CNN pre-trained by Detectron2
docker run --gpus device=1 -d -it --shm-size 32G --mount source=$(pwd),target=/home/appuser/wwtp,type=bind tumbgd/detectron2
docker exec -it <docker-container-id> bash
Specify --gpus device
according to your own machine.
-
Fix names of test images:
python fix_test.py
Note that only run ONCE.
-
Get results:
python run.py