├── README.md ├── asserts ├── .gitignore ├── inf.jpg ├── inf2.jpg └── inf3.jpg ├── ckp └── gitignore ├── config ├── __init__.py ├── coco.py └── voc.py ├── dataset ├── __init__.py ├── coco.py ├── transform.py ├── utils.py └── voc.py ├── inference.py ├── log └── .gitignore ├── loss ├── __init__.py ├── loss.py ├── losses.py └── utils.py ├── model ├── __init__.py ├── backbone │ ├── __init__.py │ └── renset.py ├── centernet.py ├── decoder.py ├── fpn.py └── head.py ├── train_voc.py └── trainer ├── __init__.py ├── lr_scheduler.py └── trainer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /asserts/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /asserts/inf.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/asserts/inf.jpg -------------------------------------------------------------------------------- /asserts/inf2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/asserts/inf2.jpg -------------------------------------------------------------------------------- /asserts/inf3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/asserts/inf3.jpg -------------------------------------------------------------------------------- /ckp/gitignore: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /config/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/config/coco.py -------------------------------------------------------------------------------- /config/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/config/voc.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/dataset/transform.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /dataset/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/dataset/voc.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/inference.py -------------------------------------------------------------------------------- /log/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/loss/loss.py -------------------------------------------------------------------------------- /loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/loss/losses.py -------------------------------------------------------------------------------- /loss/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/loss/utils.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /model/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /model/backbone/renset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/model/backbone/renset.py -------------------------------------------------------------------------------- /model/centernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/model/centernet.py -------------------------------------------------------------------------------- /model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/model/decoder.py -------------------------------------------------------------------------------- /model/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/model/fpn.py -------------------------------------------------------------------------------- /model/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/model/head.py -------------------------------------------------------------------------------- /train_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/train_voc.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /trainer/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/trainer/lr_scheduler.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gakkiri/simple-centernet-pytorch/HEAD/trainer/trainer.py --------------------------------------------------------------------------------