├── .gitignore ├── LICENSE ├── README.md ├── assets ├── demo.png ├── dog.jpg ├── git_fig.png └── logo.png ├── datasets └── README.md ├── demo └── MegEngine │ └── cpp │ ├── README.md │ ├── build.sh │ └── yolox.cpp ├── docs └── train_custom_data.md ├── exps ├── default │ ├── nano.py │ ├── yolov3.py │ ├── yolox_l.py │ ├── yolox_m.py │ ├── yolox_s.py │ ├── yolox_tiny.py │ └── yolox_x.py └── example │ └── yolox_voc │ └── yolox_voc_s.py ├── requirements.txt ├── setup.cfg ├── setup.py ├── tools ├── demo.py ├── eval.py ├── export_mge.py └── train.py └── yolox ├── __init__.py ├── core ├── __init__.py └── trainer.py ├── data ├── __init__.py ├── data_augment.py ├── data_prefetcher.py ├── dataloading.py ├── datasets │ ├── __init__.py │ ├── coco.py │ ├── coco_classes.py │ ├── datasets_wrapper.py │ ├── mosaicdetection.py │ ├── voc.py │ └── voc_classes.py └── samplers.py ├── evaluators ├── __init__.py ├── coco_evaluator.py ├── voc_eval.py └── voc_evaluator.py ├── exp ├── __init__.py ├── base_exp.py ├── build.py └── yolox_base.py ├── layers ├── __init__.py ├── csrc │ ├── cocoeval │ │ ├── cocoeval.cpp │ │ └── cocoeval.h │ └── vision.cpp └── fast_coco_eval_api.py ├── models ├── __init__.py ├── darknet.py ├── losses.py ├── network_blocks.py ├── yolo_fpn.py ├── yolo_head.py ├── yolo_pafpn.py └── yolox.py └── utils ├── __init__.py ├── boxes.py ├── checkpoint.py ├── comm.py ├── demo_utils.py ├── ema.py ├── logger.py ├── lr_scheduler.py ├── metric.py ├── model_utils.py ├── setup_env.py ├── sgd.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/README.md -------------------------------------------------------------------------------- /assets/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/assets/demo.png -------------------------------------------------------------------------------- /assets/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/assets/dog.jpg -------------------------------------------------------------------------------- /assets/git_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/assets/git_fig.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/assets/logo.png -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/datasets/README.md -------------------------------------------------------------------------------- /demo/MegEngine/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/demo/MegEngine/cpp/README.md -------------------------------------------------------------------------------- /demo/MegEngine/cpp/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/demo/MegEngine/cpp/build.sh -------------------------------------------------------------------------------- /demo/MegEngine/cpp/yolox.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/demo/MegEngine/cpp/yolox.cpp -------------------------------------------------------------------------------- /docs/train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/docs/train_custom_data.md -------------------------------------------------------------------------------- /exps/default/nano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/nano.py -------------------------------------------------------------------------------- /exps/default/yolov3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolov3.py -------------------------------------------------------------------------------- /exps/default/yolox_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolox_l.py -------------------------------------------------------------------------------- /exps/default/yolox_m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolox_m.py -------------------------------------------------------------------------------- /exps/default/yolox_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolox_s.py -------------------------------------------------------------------------------- /exps/default/yolox_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolox_tiny.py -------------------------------------------------------------------------------- /exps/default/yolox_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/default/yolox_x.py -------------------------------------------------------------------------------- /exps/example/yolox_voc/yolox_voc_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/exps/example/yolox_voc/yolox_voc_s.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/setup.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/export_mge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/tools/export_mge.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/tools/train.py -------------------------------------------------------------------------------- /yolox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/__init__.py -------------------------------------------------------------------------------- /yolox/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/core/__init__.py -------------------------------------------------------------------------------- /yolox/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/core/trainer.py -------------------------------------------------------------------------------- /yolox/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/__init__.py -------------------------------------------------------------------------------- /yolox/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/data_augment.py -------------------------------------------------------------------------------- /yolox/data/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/data_prefetcher.py -------------------------------------------------------------------------------- /yolox/data/dataloading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/dataloading.py -------------------------------------------------------------------------------- /yolox/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/__init__.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/coco.py -------------------------------------------------------------------------------- /yolox/data/datasets/coco_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/coco_classes.py -------------------------------------------------------------------------------- /yolox/data/datasets/datasets_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/datasets_wrapper.py -------------------------------------------------------------------------------- /yolox/data/datasets/mosaicdetection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/mosaicdetection.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/voc.py -------------------------------------------------------------------------------- /yolox/data/datasets/voc_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/datasets/voc_classes.py -------------------------------------------------------------------------------- /yolox/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/data/samplers.py -------------------------------------------------------------------------------- /yolox/evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/evaluators/__init__.py -------------------------------------------------------------------------------- /yolox/evaluators/coco_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/evaluators/coco_evaluator.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/evaluators/voc_eval.py -------------------------------------------------------------------------------- /yolox/evaluators/voc_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/evaluators/voc_evaluator.py -------------------------------------------------------------------------------- /yolox/exp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/exp/__init__.py -------------------------------------------------------------------------------- /yolox/exp/base_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/exp/base_exp.py -------------------------------------------------------------------------------- /yolox/exp/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/exp/build.py -------------------------------------------------------------------------------- /yolox/exp/yolox_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/exp/yolox_base.py -------------------------------------------------------------------------------- /yolox/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/layers/__init__.py -------------------------------------------------------------------------------- /yolox/layers/csrc/cocoeval/cocoeval.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/layers/csrc/cocoeval/cocoeval.cpp -------------------------------------------------------------------------------- /yolox/layers/csrc/cocoeval/cocoeval.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/layers/csrc/cocoeval/cocoeval.h -------------------------------------------------------------------------------- /yolox/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /yolox/layers/fast_coco_eval_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/layers/fast_coco_eval_api.py -------------------------------------------------------------------------------- /yolox/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/__init__.py -------------------------------------------------------------------------------- /yolox/models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/darknet.py -------------------------------------------------------------------------------- /yolox/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/losses.py -------------------------------------------------------------------------------- /yolox/models/network_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/network_blocks.py -------------------------------------------------------------------------------- /yolox/models/yolo_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/yolo_fpn.py -------------------------------------------------------------------------------- /yolox/models/yolo_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/yolo_head.py -------------------------------------------------------------------------------- /yolox/models/yolo_pafpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/yolo_pafpn.py -------------------------------------------------------------------------------- /yolox/models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/models/yolox.py -------------------------------------------------------------------------------- /yolox/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/__init__.py -------------------------------------------------------------------------------- /yolox/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/boxes.py -------------------------------------------------------------------------------- /yolox/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/checkpoint.py -------------------------------------------------------------------------------- /yolox/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/comm.py -------------------------------------------------------------------------------- /yolox/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/demo_utils.py -------------------------------------------------------------------------------- /yolox/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/ema.py -------------------------------------------------------------------------------- /yolox/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/logger.py -------------------------------------------------------------------------------- /yolox/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/lr_scheduler.py -------------------------------------------------------------------------------- /yolox/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/metric.py -------------------------------------------------------------------------------- /yolox/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/model_utils.py -------------------------------------------------------------------------------- /yolox/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/setup_env.py -------------------------------------------------------------------------------- /yolox/utils/sgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/sgd.py -------------------------------------------------------------------------------- /yolox/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MegEngine/YOLOX/HEAD/yolox/utils/visualize.py --------------------------------------------------------------------------------