• Stars
    star
    166
  • Rank 227,748 (Top 5 %)
  • Language
    Python
  • License
    MIT License
  • Created about 5 years ago
  • Updated over 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Instance Segmentation based on CenterNet and CondInst

CenterNet-CondInst

CenterNet: Objects as Points + CondInst: Conditional Convolutions for Instance Segmentation

Installation

Please refer to CnterNet INSTALL.md for installation instructions.

Training

## note : seg_weight default setting is 1. You can set it to other value to get better performance.
cd src
python main.py ctseg --exp_id coco_dla_1x --batch_size 20 --master_batch 9 --lr 1.25e-4 --gpus 0,1 --num_workers 4

Eval

## not support flip test and multi scale test
cd src
python test.py ctseg --exp_id coco_dla_1x --keep_res --resume

Visualization

cd src
python demo.py ctseg --exp_id coco_dla_1x --keep_res --resume --demo ../data/coco/val2017

Result

type AP AP50 AP75 APs APm APl
box 0.358 0.540 0.384 0.154 0.391 0.535
mask 0.306 0.493 0.317 0.100 0.341 0.490

backbone=dla_34, batch=32

Reference

  1. CenterNet
  2. CondInst