├── LICENSE ├── README.md ├── VisDrone_Dataset.py ├── annotations └── VisDrone2019-DET_val_coco.json ├── configs └── yolc.py ├── dist_train.sh ├── eval_yolc.py ├── framework.jpg ├── gen_crop.py ├── inference_YOLC.py ├── models ├── __init__.py ├── backbones │ ├── __init__.py │ └── vit.py ├── dense_heads │ ├── __init__.py │ └── yolc_head.py ├── detectors │ ├── __init__.py │ └── yolc.py └── losses │ ├── __init__.py │ └── gwd_loss.py ├── test.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/README.md -------------------------------------------------------------------------------- /VisDrone_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/VisDrone_Dataset.py -------------------------------------------------------------------------------- /annotations/VisDrone2019-DET_val_coco.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/annotations/VisDrone2019-DET_val_coco.json -------------------------------------------------------------------------------- /configs/yolc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/configs/yolc.py -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/dist_train.sh -------------------------------------------------------------------------------- /eval_yolc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/eval_yolc.py -------------------------------------------------------------------------------- /framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/framework.jpg -------------------------------------------------------------------------------- /gen_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/gen_crop.py -------------------------------------------------------------------------------- /inference_YOLC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/inference_YOLC.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | from .vit import ViT 2 | 3 | __all__ = ['ViT'] -------------------------------------------------------------------------------- /models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/backbones/vit.py -------------------------------------------------------------------------------- /models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /models/dense_heads/yolc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/dense_heads/yolc_head.py -------------------------------------------------------------------------------- /models/detectors/__init__.py: -------------------------------------------------------------------------------- 1 | from .yolc import YOLC -------------------------------------------------------------------------------- /models/detectors/yolc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/detectors/yolc.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .gwd_loss import GWDLoss 2 | -------------------------------------------------------------------------------- /models/losses/gwd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/models/losses/gwd_loss.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dawn-ech/YOLC/HEAD/train.py --------------------------------------------------------------------------------