├── LICENSE ├── README.md ├── lib ├── __init__.py ├── config.py ├── dataloader │ ├── __init__.py │ ├── augmentations.py │ └── dataloader.py ├── model │ ├── __init__.py │ ├── anchors.py │ ├── aploss.py │ └── model.py └── util │ ├── __init__.py │ ├── calc_iou.py │ ├── coco_eval.py │ ├── pascal_voc_eval.py │ ├── utils.py │ └── voc_eval.py ├── test.py ├── test.sh ├── train.py └── train.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/README.md -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/config.py -------------------------------------------------------------------------------- /lib/dataloader/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /lib/dataloader/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/dataloader/augmentations.py -------------------------------------------------------------------------------- /lib/dataloader/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/dataloader/dataloader.py -------------------------------------------------------------------------------- /lib/model/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /lib/model/anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/model/anchors.py -------------------------------------------------------------------------------- /lib/model/aploss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/model/aploss.py -------------------------------------------------------------------------------- /lib/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/model/model.py -------------------------------------------------------------------------------- /lib/util/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /lib/util/calc_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/util/calc_iou.py -------------------------------------------------------------------------------- /lib/util/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/util/coco_eval.py -------------------------------------------------------------------------------- /lib/util/pascal_voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/util/pascal_voc_eval.py -------------------------------------------------------------------------------- /lib/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/util/utils.py -------------------------------------------------------------------------------- /lib/util/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/lib/util/voc_eval.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/test.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/test.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cccorn/AP-loss/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- 1 | python train.py --dataset coco 2 | --------------------------------------------------------------------------------