├── .flake8 ├── .gitignore ├── .isort.cfg ├── .pre-commit-config.yaml ├── Dockerfile ├── LICENSE.txt ├── README.md ├── assets └── logo.png ├── bash_scripts ├── convert_single_models.sh ├── detect.sh ├── evolve.sh ├── train_voc_obj365.sh ├── train_voc_obj365_animals.sh ├── train_voc_obj365_animals_tableware.sh └── val.sh ├── cerberusdet ├── __init__.py ├── convert_to_cerber.py ├── data │ ├── __init__.py │ ├── augmentations.py │ ├── dataloaders.py │ ├── dataset_images.py │ ├── datasets.py │ └── samplers.py ├── detect.py ├── evolvers │ ├── __init__.py │ ├── base_evolver.py │ ├── checkpoint_logger.py │ ├── file_logger.py │ ├── predefined_evolvers.py │ ├── ray_evolver.py │ └── yolov5_evolver.py ├── models │ ├── cerberus.py │ ├── common.py │ ├── experimental.py │ ├── yolo.py │ ├── yolov8x.yaml │ ├── yolov8x_voc_obj365.yaml │ ├── yolov8x_voc_obj365_animals.yaml │ └── yolov8x_voc_obj365_animals_tableware.yaml ├── strip_weights.py ├── train.py ├── trainers │ ├── __init__.py │ ├── averaging.py │ └── base_trainer.py ├── utils │ ├── checks.py │ ├── ckpt_utils.py │ ├── general.py │ ├── loss.py │ ├── metrics.py │ ├── mlflow_logging.py │ ├── models_manager.py │ ├── plots.py │ ├── tal.py │ ├── torch_utils.py │ └── train_utils.py └── val.py ├── data ├── hyps │ ├── hyp.cerber-voc_obj365.yaml │ └── hyp.cerber-voc_obj365_subsets.yaml ├── images │ ├── image1.png │ ├── image2.jpg │ ├── image3.jpg │ ├── image4.jpg │ └── image5.jpeg ├── scripts │ ├── objects365_part.py │ └── voc.py ├── voc_obj365_animals.yaml ├── voc_obj365_animals_tableware.yaml └── voc_obj365_full.yaml ├── docker-compose.yml ├── mypy.ini └── pyproject.toml /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/README.md -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/assets/logo.png -------------------------------------------------------------------------------- /bash_scripts/convert_single_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/convert_single_models.sh -------------------------------------------------------------------------------- /bash_scripts/detect.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/detect.sh -------------------------------------------------------------------------------- /bash_scripts/evolve.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/evolve.sh -------------------------------------------------------------------------------- /bash_scripts/train_voc_obj365.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/train_voc_obj365.sh -------------------------------------------------------------------------------- /bash_scripts/train_voc_obj365_animals.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/train_voc_obj365_animals.sh -------------------------------------------------------------------------------- /bash_scripts/train_voc_obj365_animals_tableware.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/train_voc_obj365_animals_tableware.sh -------------------------------------------------------------------------------- /bash_scripts/val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/bash_scripts/val.sh -------------------------------------------------------------------------------- /cerberusdet/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.0.0" 2 | -------------------------------------------------------------------------------- /cerberusdet/convert_to_cerber.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/convert_to_cerber.py -------------------------------------------------------------------------------- /cerberusdet/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cerberusdet/data/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/data/augmentations.py -------------------------------------------------------------------------------- /cerberusdet/data/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/data/dataloaders.py -------------------------------------------------------------------------------- /cerberusdet/data/dataset_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/data/dataset_images.py -------------------------------------------------------------------------------- /cerberusdet/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/data/datasets.py -------------------------------------------------------------------------------- /cerberusdet/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/data/samplers.py -------------------------------------------------------------------------------- /cerberusdet/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/detect.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/__init__.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/base_evolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/base_evolver.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/checkpoint_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/checkpoint_logger.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/file_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/file_logger.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/predefined_evolvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/predefined_evolvers.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/ray_evolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/ray_evolver.py -------------------------------------------------------------------------------- /cerberusdet/evolvers/yolov5_evolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/evolvers/yolov5_evolver.py -------------------------------------------------------------------------------- /cerberusdet/models/cerberus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/cerberus.py -------------------------------------------------------------------------------- /cerberusdet/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/common.py -------------------------------------------------------------------------------- /cerberusdet/models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/experimental.py -------------------------------------------------------------------------------- /cerberusdet/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/yolo.py -------------------------------------------------------------------------------- /cerberusdet/models/yolov8x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/yolov8x.yaml -------------------------------------------------------------------------------- /cerberusdet/models/yolov8x_voc_obj365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/yolov8x_voc_obj365.yaml -------------------------------------------------------------------------------- /cerberusdet/models/yolov8x_voc_obj365_animals.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/yolov8x_voc_obj365_animals.yaml -------------------------------------------------------------------------------- /cerberusdet/models/yolov8x_voc_obj365_animals_tableware.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/models/yolov8x_voc_obj365_animals_tableware.yaml -------------------------------------------------------------------------------- /cerberusdet/strip_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/strip_weights.py -------------------------------------------------------------------------------- /cerberusdet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/train.py -------------------------------------------------------------------------------- /cerberusdet/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cerberusdet/trainers/averaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/trainers/averaging.py -------------------------------------------------------------------------------- /cerberusdet/trainers/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/trainers/base_trainer.py -------------------------------------------------------------------------------- /cerberusdet/utils/checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/checks.py -------------------------------------------------------------------------------- /cerberusdet/utils/ckpt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/ckpt_utils.py -------------------------------------------------------------------------------- /cerberusdet/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/general.py -------------------------------------------------------------------------------- /cerberusdet/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/loss.py -------------------------------------------------------------------------------- /cerberusdet/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/metrics.py -------------------------------------------------------------------------------- /cerberusdet/utils/mlflow_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/mlflow_logging.py -------------------------------------------------------------------------------- /cerberusdet/utils/models_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/models_manager.py -------------------------------------------------------------------------------- /cerberusdet/utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/plots.py -------------------------------------------------------------------------------- /cerberusdet/utils/tal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/tal.py -------------------------------------------------------------------------------- /cerberusdet/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/torch_utils.py -------------------------------------------------------------------------------- /cerberusdet/utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/utils/train_utils.py -------------------------------------------------------------------------------- /cerberusdet/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/cerberusdet/val.py -------------------------------------------------------------------------------- /data/hyps/hyp.cerber-voc_obj365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/hyps/hyp.cerber-voc_obj365.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.cerber-voc_obj365_subsets.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/hyps/hyp.cerber-voc_obj365_subsets.yaml -------------------------------------------------------------------------------- /data/images/image1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/images/image1.png -------------------------------------------------------------------------------- /data/images/image2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/images/image2.jpg -------------------------------------------------------------------------------- /data/images/image3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/images/image3.jpg -------------------------------------------------------------------------------- /data/images/image4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/images/image4.jpg -------------------------------------------------------------------------------- /data/images/image5.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/images/image5.jpeg -------------------------------------------------------------------------------- /data/scripts/objects365_part.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/scripts/objects365_part.py -------------------------------------------------------------------------------- /data/scripts/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/scripts/voc.py -------------------------------------------------------------------------------- /data/voc_obj365_animals.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/voc_obj365_animals.yaml -------------------------------------------------------------------------------- /data/voc_obj365_animals_tableware.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/voc_obj365_animals_tableware.yaml -------------------------------------------------------------------------------- /data/voc_obj365_full.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/data/voc_obj365_full.yaml -------------------------------------------------------------------------------- /docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/docker-compose.yml -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ai-forever/CerberusDet/HEAD/pyproject.toml --------------------------------------------------------------------------------