detectron2 for PubLayNet
- Document layout analysis using detectron2 framework, this is the repos of PubLayNet
python3 setup.py install
- document layout analysis
python3 demo/demo.py --config-file configs/DLA_mask_rcnn_X_101_32x8d_FPN_3x.yaml --input test.png --output ./ --confidence-threshold 0.5 --opts MODEL.WEIGHTS model_final_trimmed.pth MODEL.DEVICE cpu
 |
 |
|---|---|
 |
 |
 |
 |
- object detection by faster-rcnn
python3 demo/demo.py --config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml --input test.jpg --output ./ --opts MODEL.WEIGHTS model_final_faster_rcnn.pkl MODEL.DEVICE cpu
 |
|---|
 |
- object detection by mask-rcnn
python3 demo/demo.py --config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml --input test.jpg --output ./ --opts MODEL.WEIGHTS model_final_mask_rcnn.pkl MODEL.DEVICE cpu
 |
|---|
 |
- human joint keypoint detection
python3 demo/demo.py --config-file configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x.yaml --input test.jpg --output ./ --opts MODEL.WEIGHTS model_final_keypoint_rcnn.pkl MODEL.DEVICE cpu
 |
|---|
 |
- more models can be found in MODEL_ZOO