Skip to content

phungpx/maskRCNN_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

57 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MaskRCNN

Todo

Usage

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/...

Papers

  1. Fast RCNN
  2. Faster RCNN: Towards Real-Time Object Detection with Region Proposal Networks
  3. Mask RCNN

Documents

  1. The best explanation, I've ever read.
https://www.telesens.co/2018/03/11/object-detection-and-classification-using-r-cnns/
  1. Pytorch Setting.
https://pytorch.org/vision/stable/models.html#object-detection-instance-segmentation-and-person-keypoint-detection
  1. 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
  1. Non-maxmimum Suppression (NMS)
https://learnopencv.com/non-maximum-suppression-theory-and-implementation-in-pytorch/
  1. mAP
https://github.com/rafaelpadilla/Object-Detection-Metrics