Remove defgrid mask head by commenting out the add_defgrid_maskhead_config(cfg)
from the def setup(args)
in train_net_palmira.py
python train_palmira.py \
--config-file configs/dconv/dconv_c3-c5.yaml \
--num-gpus 4
To toggle other layers of the ResNet, edit this section of the config.yaml
MODEL:
RESNETS:
DEFORM_ON_PER_STAGE: [ False, True, True, True ] # on Res3, Res4, Res5
Remove defgrid mask head by commenting out the add_defgrid_maskhead_config(cfg)
from the def setup(args)
in train_net_palmira.py
python train_palmira.py \
--config-file configs/mrcnn/vanilla_mrcnn.yaml \
--num-gpus 4
Please refer to respective folders for code, train scripts, configs, and the README for usage and installation instructions.
Model Name | Official Code Repo |
---|---|
CondInst | Code |
PointRend | Code |
BMRCNN | Code |
DETR | Code |
@misc{carion2020endtoend,
title={End-to-End Object Detection with Transformers},
author={Nicolas Carion and Francisco Massa and Gabriel Synnaeve and Nicolas Usunier and Alexander Kirillov and Sergey Zagoruyko},
year={2020},
eprint={2005.12872},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
@inproceedings{tian2020conditional,
title = {Conditional Convolutions for Instance Segmentation},
author = {Tian, Zhi and Shen, Chunhua and Chen, Hao},
booktitle = {Proc. Eur. Conf. Computer Vision (ECCV)},
year = {2020}
}
@article{ChengWHL20,
title={Boundary-preserving Mask R-CNN},
author={Tianheng Cheng and Xinggang Wang and Lichao Huang and Wenyu Liu},
booktitle={ECCV},
year={2020}
}
@InProceedings{kirillov2019pointrend,
title={{PointRend}: Image Segmentation as Rendering},
author={Alexander Kirillov and Yuxin Wu and Kaiming He and Ross Girshick},
journal={ArXiv:1912.08193},
year={2019}
}