- Mask RCNN Resnet 50 FPN
- Mask RCNN MobilenetV3 Large FPN
- Mask RCNN MobilenetV2
- Faster RCNN Resnet 50 FPN
- Faster RCNN MobilenetV3 FPN
- mAP
- COCO Evaluator
- Support Labelme format
- Support Altheia format
- Support Pascal VOC format
Training
CUDA_VISIBLE_DEVICE=0,1,2,3 python main.py --config configs/... --num-epochs <x> --gpu-indices 0,1,2,3
Testing
CUDA_VISIBLE_DEVICE=0,1,2,3 python -m flame configs/...
- The best explanation, I've ever read.
https://www.telesens.co/2018/03/11/object-detection-and-classification-using-r-cnns/
- Pytorch Setting.
https://pytorch.org/vision/stable/models.html#object-detection-instance-segmentation-and-person-keypoint-detection
- RoI Align vs RoI Pooling
- RoI Pooling
https://erdem.pl/2020/02/understanding-region-of-interest-ro-i-pooling
- RoI Align
https://erdem.pl/2020/02/understanding-region-of-interest-part-2-ro-i-align
- Non-maxmimum Suppression (NMS)
https://learnopencv.com/non-maximum-suppression-theory-and-implementation-in-pytorch/
- mAP
https://github.com/rafaelpadilla/Object-Detection-Metrics