facebookresearch / facebookresearch/detectron2

Add GAN Mask R-CNN model to detectron2 model zoo

Open
#3,829 0 comments 3 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
34.7k
Forks
7.9k
PR merge metrics
No merged PRs in 30d

Description

## 🚀 Feature
Add GAN Mask R-CNN model to detectron2 model zoo

## Motivation & Examples

Tell us why the feature is useful.
Some research has been done that demonstrates GAN Mask R-CNN is able to produce crisper edges compared to the non-GAN version.
- See [Instance Semantic Segmentation Benefits from Generative Adversarial Networks](https://arxiv.org/abs/2010.13757) for research paper.
- See [GAN_Mask-RCNN](https://github.com/quangle2110/GAN_Mask-RCNN) for code from research paper.

Describe what the feature would look like, if it is implemented.
```python
model_file = "COCO-InstanceSegmentation/gan_mask_rcnn_R_50_FPN.yaml"
cfg = get_cfg()
cfg.merge_from_file(model_zoo.get_config_file(model_file))

# omitting dataset registration and other cfg parameters
trainer = DefaultTrainer(cfg)
trainer.resume_or_load(resume=False)
trainer.train()
```

## Note

We only consider adding new features if they are relevant to many users.

If you request implementation of research papers -- we only consider papers that have enough significance and prevalance in the object detection field.

We do not take requests for most projects in the `projects/` directory, because they are research code release that is mainly for other researchers to reproduce results.

"Make X faster/accurate" is not a valid feature request. "Implement a concrete feature that can make X faster/accurate" can be a valid feature request.

Instead of adding features inside detectron2,
you can implement many features by [extending detectron2](https://detectron2.readthedocs.io/tutorials/extend.html).
The [projects/](https://github.com/facebookresearch/detectron2/tree/main/projects/) directory contains many of such examples.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.