├── .clang-format ├── .dockerignore ├── .flake8 ├── .gitattributes ├── .github └── workflows │ └── py.yaml ├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── benchmarks.py ├── classify ├── predict.py ├── train.py ├── tutorial.ipynb └── val.py ├── data ├── Argoverse.yaml ├── GlobalWheat2020.yaml ├── ImageNet.yaml ├── Objects365.yaml ├── SKU-110K.yaml ├── VOC.yaml ├── VisDrone.yaml ├── coco.yaml ├── coco128-seg.yaml ├── coco128.yaml ├── hyps │ ├── hyp.Objects365.yaml │ ├── hyp.VOC.yaml │ ├── hyp.no-augmentation.yaml │ ├── hyp.scratch-high.yaml │ ├── hyp.scratch-low.yaml │ └── hyp.scratch-med.yaml ├── images │ ├── bus.jpg │ └── zidane.jpg ├── scripts │ ├── download_weights.sh │ ├── get_coco.sh │ ├── get_coco128.sh │ └── get_imagenet.sh └── xView.yaml ├── detect.py ├── dev ├── README.md ├── dev-requirements.txt ├── linter.sh └── run_unittest.sh ├── document └── README.md ├── export.py ├── hubconf.py ├── models ├── __init__.py ├── common.py ├── experimental.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-bifpn.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p34.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5n6.yaml │ ├── yolov5s-LeakyReLU.yaml │ ├── yolov5s-ghost.yaml │ ├── yolov5s-transformer.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── segment │ ├── yolov5l-seg.yaml │ ├── yolov5m-seg.yaml │ ├── yolov5n-seg.yaml │ ├── yolov5s-seg.yaml │ └── yolov5x-seg.yaml ├── tf.py ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5n.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── oneflow_train_test ├── READEME.md ├── config.sh ├── train_all_test.sh ├── train_cls_test.sh ├── train_default_test.sh └── train_seg_test.sh ├── requirements.txt ├── segment ├── predict.py ├── train.py ├── tutorial.ipynb └── val.py ├── setup.cfg ├── steup.cfg ├── train.py ├── tutorial.ipynb ├── utils ├── __init__.py ├── activations.py ├── augmentations.py ├── autoanchor.py ├── autobatch.py ├── callbacks.py ├── dataloaders.py ├── downloads.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── loggers │ ├── __init__.py │ ├── clearml │ │ ├── README.md │ │ ├── __init__.py │ │ ├── clearml_utils.py │ │ └── hpo.py │ ├── comet │ │ ├── README.md │ │ ├── __init__.py │ │ ├── comet_utils.py │ │ └── hpo.py │ └── wandb │ │ ├── README.md │ │ ├── __init__.py │ │ ├── log_dataset.py │ │ ├── sweep.py │ │ ├── sweep.yaml │ │ └── wandb_utils.py ├── loss.py ├── metrics.py ├── plots.py ├── segment │ ├── __init__.py │ ├── augmentations.py │ ├── dataloaders.py │ ├── general.py │ ├── loss.py │ ├── metrics.py │ └── plots.py ├── temp_repair_tool.py ├── torch_utils.py └── triton.py └── val.py /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.clang-format -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.dockerignore -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/py.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.github/workflows/py.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/benchmarks.py -------------------------------------------------------------------------------- /classify/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/classify/predict.py -------------------------------------------------------------------------------- /classify/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/classify/train.py -------------------------------------------------------------------------------- /classify/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/classify/tutorial.ipynb -------------------------------------------------------------------------------- /classify/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/classify/val.py -------------------------------------------------------------------------------- /data/Argoverse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/Argoverse.yaml -------------------------------------------------------------------------------- /data/GlobalWheat2020.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/GlobalWheat2020.yaml -------------------------------------------------------------------------------- /data/ImageNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/ImageNet.yaml -------------------------------------------------------------------------------- /data/Objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/Objects365.yaml -------------------------------------------------------------------------------- /data/SKU-110K.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/SKU-110K.yaml -------------------------------------------------------------------------------- /data/VOC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/VOC.yaml -------------------------------------------------------------------------------- /data/VisDrone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/VisDrone.yaml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/coco128-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/coco128-seg.yaml -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.Objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.Objects365.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.VOC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.VOC.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.no-augmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.no-augmentation.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-high.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.scratch-high.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-low.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.scratch-low.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-med.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/hyps/hyp.scratch-med.yaml -------------------------------------------------------------------------------- /data/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/images/bus.jpg -------------------------------------------------------------------------------- /data/images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/images/zidane.jpg -------------------------------------------------------------------------------- /data/scripts/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/scripts/download_weights.sh -------------------------------------------------------------------------------- /data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /data/scripts/get_coco128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/scripts/get_coco128.sh -------------------------------------------------------------------------------- /data/scripts/get_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/scripts/get_imagenet.sh -------------------------------------------------------------------------------- /data/xView.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/data/xView.yaml -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/detect.py -------------------------------------------------------------------------------- /dev/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/dev/README.md -------------------------------------------------------------------------------- /dev/dev-requirements.txt: -------------------------------------------------------------------------------- 1 | flake8==3.8.1 2 | isort==5.10.1 3 | black==21.4b2 4 | -------------------------------------------------------------------------------- /dev/linter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/dev/linter.sh -------------------------------------------------------------------------------- /dev/run_unittest.sh: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /document/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/document/README.md -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/export.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-bifpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p34.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-p34.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5n6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5n6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-LeakyReLU.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5s-LeakyReLU.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-ghost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5s-ghost.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5s-transformer.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/segment/yolov5l-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/segment/yolov5l-seg.yaml -------------------------------------------------------------------------------- /models/segment/yolov5m-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/segment/yolov5m-seg.yaml -------------------------------------------------------------------------------- /models/segment/yolov5n-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/segment/yolov5n-seg.yaml -------------------------------------------------------------------------------- /models/segment/yolov5s-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/segment/yolov5s-seg.yaml -------------------------------------------------------------------------------- /models/segment/yolov5x-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/segment/yolov5x-seg.yaml -------------------------------------------------------------------------------- /models/tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/tf.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolov5n.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /oneflow_train_test/READEME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/READEME.md -------------------------------------------------------------------------------- /oneflow_train_test/config.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/config.sh -------------------------------------------------------------------------------- /oneflow_train_test/train_all_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/train_all_test.sh -------------------------------------------------------------------------------- /oneflow_train_test/train_cls_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/train_cls_test.sh -------------------------------------------------------------------------------- /oneflow_train_test/train_default_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/train_default_test.sh -------------------------------------------------------------------------------- /oneflow_train_test/train_seg_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/oneflow_train_test/train_seg_test.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/requirements.txt -------------------------------------------------------------------------------- /segment/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/segment/predict.py -------------------------------------------------------------------------------- /segment/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/segment/train.py -------------------------------------------------------------------------------- /segment/tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/segment/tutorial.ipynb -------------------------------------------------------------------------------- /segment/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/segment/val.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/setup.cfg -------------------------------------------------------------------------------- /steup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/steup.cfg -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/train.py -------------------------------------------------------------------------------- /tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/tutorial.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/dataloaders.py -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/clearml/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/clearml/README.md -------------------------------------------------------------------------------- /utils/loggers/clearml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/clearml/clearml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/clearml/clearml_utils.py -------------------------------------------------------------------------------- /utils/loggers/clearml/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/clearml/hpo.py -------------------------------------------------------------------------------- /utils/loggers/comet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/comet/README.md -------------------------------------------------------------------------------- /utils/loggers/comet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/comet/__init__.py -------------------------------------------------------------------------------- /utils/loggers/comet/comet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/comet/comet_utils.py -------------------------------------------------------------------------------- /utils/loggers/comet/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/comet/hpo.py -------------------------------------------------------------------------------- /utils/loggers/wandb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/wandb/README.md -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/wandb/log_dataset.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/wandb/sweep.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/wandb/sweep.yaml -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/segment/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/segment/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/augmentations.py -------------------------------------------------------------------------------- /utils/segment/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/dataloaders.py -------------------------------------------------------------------------------- /utils/segment/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/general.py -------------------------------------------------------------------------------- /utils/segment/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/loss.py -------------------------------------------------------------------------------- /utils/segment/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/metrics.py -------------------------------------------------------------------------------- /utils/segment/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/segment/plots.py -------------------------------------------------------------------------------- /utils/temp_repair_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/temp_repair_tool.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/utils/triton.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/one-yolov5/HEAD/val.py --------------------------------------------------------------------------------