https://github.com/acecoooool/retinanet
Implementation of RetinaNet (converted from maskrcnn-benchmark)
https://github.com/acecoooool/retinanet
pytorch retinanet
Last synced: about 1 month ago
JSON representation
Implementation of RetinaNet (converted from maskrcnn-benchmark)
- Host: GitHub
- URL: https://github.com/acecoooool/retinanet
- Owner: AceCoooool
- License: mit
- Created: 2019-05-15T11:48:10.000Z (about 7 years ago)
- Default Branch: master
- Last Pushed: 2019-05-23T13:55:08.000Z (about 7 years ago)
- Last Synced: 2025-06-03T15:39:37.366Z (about 1 year ago)
- Topics: pytorch, retinanet
- Language: Python
- Homepage:
- Size: 1.53 MB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# RetinaNet
Implementation of RetinaNet (this is converted from [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark))
## Environment
- Python 3.6
- PyTorch 1.1
- OpenCV
## Performance
| Model | maskrcnn | converted | train(TODO) |
| :--------------------: | :------: | :----------------------------------------------------------: | :---------: |
| RetinaNet_R-50-FPN_1x | 36.3 | [36.3](https://drive.google.com/open?id=1edT78Boy6SA3EdSKvJB9sR72KShSIg2e) | |
| RetinaNet_R-101-FPN_1x | 38.5 | [38.5](https://drive.google.com/open?id=1Qrwx4PaxQJHnV_R2H8HV8QK0P6IyTi9a) | |
## Demo
```shell
$ cd scripts
$ python demo.py [--config-file ../configs/retina_resnet50_v1b_coco.yaml] [--images ../png/biking.jpg]
```
## Eval
```shell
$ cd scripts
$ python eval.py [--config-file ../configs/retina_resnet50_v1b_coco.yaml]
```
## Train
```shell
$ cd scripts
$ export NGPUS=4
$ python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py [--config-file ../configs/retina_resnet50_v1b_coco.yaml] [--skip-test false]
```
> You can modify training and test setting in `configs/xxx.yaml` and `configs/defaults.py`
## TODO
- [ ] Add freeze backbone
- [ ] Add training results