├── .pre-commit-config.yaml ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── data ├── Argoverse.yaml ├── GlobalWheat2020.yaml ├── Objects365.yaml ├── SKU-110K.yaml ├── VOC.yaml ├── VisDrone.yaml ├── coco.yaml ├── coco128.yaml ├── hyps │ ├── hyp.Objects365.yaml │ ├── hyp.VOC.yaml │ ├── hyp.scratch-high.yaml │ ├── hyp.scratch-low.yaml │ └── hyp.scratch-med.yaml ├── images │ ├── bus.jpg │ └── zidane.jpg ├── my_person.yaml ├── scripts │ ├── download_weights.sh │ ├── get_coco.sh │ └── get_coco128.sh └── xView.yaml ├── datasets └── mydata │ ├── Annotations │ ├── 1066405,1b8000ef60354f.xml │ ├── 1066405,2a72f000f214d26a.xml │ ├── 1066405,2ac2400079a6d80f.xml │ ├── 1066405,2b6000ffe20c07.xml │ ├── 1066405,2bf6e00075455d3c.xml │ ├── 1066405,2bfbf000c47880b7.xml │ ├── 1066405,2c8c7000530eb0e7.xml │ ├── 1066405,2cdca0006185e7eb.xml │ ├── 1066405,2d2c6000adf6f6f4.xml │ └── 1066405,2d6f2000fec9dcab.xml │ ├── images │ ├── 1066405,1b8000ef60354f.jpg │ ├── 1066405,2a72f000f214d26a.jpg │ ├── 1066405,2ac2400079a6d80f.jpg │ ├── 1066405,2b6000ffe20c07.jpg │ ├── 1066405,2bf6e00075455d3c.jpg │ ├── 1066405,2bfbf000c47880b7.jpg │ ├── 1066405,2c8c7000530eb0e7.jpg │ ├── 1066405,2cdca0006185e7eb.jpg │ ├── 1066405,2d2c6000adf6f6f4.jpg │ └── 1066405,2d6f2000fec9dcab.jpg │ ├── labels │ ├── 1066405,1b8000ef60354f.txt │ ├── 1066405,2a72f000f214d26a.txt │ ├── 1066405,2ac2400079a6d80f.txt │ ├── 1066405,2b6000ffe20c07.txt │ ├── 1066405,2bf6e00075455d3c.txt │ ├── 1066405,2bfbf000c47880b7.txt │ ├── 1066405,2c8c7000530eb0e7.txt │ ├── 1066405,2cdca0006185e7eb.txt │ ├── 1066405,2d2c6000adf6f6f4.txt │ └── 1066405,2d6f2000fec9dcab.txt │ ├── test.txt │ ├── train.cache │ ├── train.txt │ ├── val.cache │ └── val.txt ├── detect.py ├── export.py ├── hubconf.py ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── common.cpython-37.pyc │ ├── experimental.cpython-37.pyc │ └── yolo.cpython-37.pyc ├── 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-ghost.yaml │ ├── yolov5s-transformer.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── tf.py ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5m_C3CA.yaml ├── yolov5m_C3CBAM.yaml ├── yolov5m_C3ECA.yaml ├── yolov5m_C3SE.yaml ├── yolov5m_CBAM.yaml ├── yolov5m_CoordAtt.yaml ├── yolov5m_ECA.yaml ├── yolov5m_SE.yaml ├── yolov5n.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── requirements.txt ├── setup.cfg ├── split.py ├── train.py ├── tutorial.ipynb ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── augmentations.cpython-37.pyc │ ├── autoanchor.cpython-37.pyc │ ├── autobatch.cpython-37.pyc │ ├── callbacks.cpython-37.pyc │ ├── datasets.cpython-37.pyc │ ├── downloads.cpython-37.pyc │ ├── general.cpython-37.pyc │ ├── loss.cpython-37.pyc │ ├── metrics.cpython-37.pyc │ ├── plots.cpython-37.pyc │ └── torch_utils.cpython-37.pyc ├── activations.py ├── augmentations.py ├── autoanchor.py ├── autobatch.py ├── aws │ ├── __init__.py │ ├── mime.sh │ ├── resume.py │ └── userdata.sh ├── benchmarks.py ├── callbacks.py ├── datasets.py ├── docker │ ├── .dockerignore │ ├── Dockerfile │ └── Dockerfile-cpu ├── downloads.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── loggers │ ├── __init__.py │ ├── __pycache__ │ │ └── __init__.cpython-37.pyc │ └── wandb │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ └── wandb_utils.cpython-37.pyc │ │ ├── log_dataset.py │ │ ├── sweep.py │ │ ├── sweep.yaml │ │ └── wandb_utils.py ├── loss.py ├── metrics.py ├── plots.py └── torch_utils.py ├── val.py └── voc_label.py /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/README.md -------------------------------------------------------------------------------- /data/Argoverse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/Argoverse.yaml -------------------------------------------------------------------------------- /data/GlobalWheat2020.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/GlobalWheat2020.yaml -------------------------------------------------------------------------------- /data/Objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/Objects365.yaml -------------------------------------------------------------------------------- /data/SKU-110K.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/SKU-110K.yaml -------------------------------------------------------------------------------- /data/VOC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/VOC.yaml -------------------------------------------------------------------------------- /data/VisDrone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/VisDrone.yaml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.Objects365.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/hyps/hyp.Objects365.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.VOC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/hyps/hyp.VOC.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-high.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/hyps/hyp.scratch-high.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-low.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/hyps/hyp.scratch-low.yaml -------------------------------------------------------------------------------- /data/hyps/hyp.scratch-med.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/hyps/hyp.scratch-med.yaml -------------------------------------------------------------------------------- /data/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/images/bus.jpg -------------------------------------------------------------------------------- /data/images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/images/zidane.jpg -------------------------------------------------------------------------------- /data/my_person.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/my_person.yaml -------------------------------------------------------------------------------- /data/scripts/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/scripts/download_weights.sh -------------------------------------------------------------------------------- /data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /data/scripts/get_coco128.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/scripts/get_coco128.sh -------------------------------------------------------------------------------- /data/xView.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/data/xView.yaml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,1b8000ef60354f.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,1b8000ef60354f.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2a72f000f214d26a.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2a72f000f214d26a.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2ac2400079a6d80f.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2ac2400079a6d80f.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2b6000ffe20c07.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2b6000ffe20c07.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2bf6e00075455d3c.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2bf6e00075455d3c.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2bfbf000c47880b7.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2bfbf000c47880b7.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2c8c7000530eb0e7.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2c8c7000530eb0e7.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2cdca0006185e7eb.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2cdca0006185e7eb.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2d2c6000adf6f6f4.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2d2c6000adf6f6f4.xml -------------------------------------------------------------------------------- /datasets/mydata/Annotations/1066405,2d6f2000fec9dcab.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/Annotations/1066405,2d6f2000fec9dcab.xml -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,1b8000ef60354f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,1b8000ef60354f.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2a72f000f214d26a.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2a72f000f214d26a.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2ac2400079a6d80f.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2ac2400079a6d80f.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2b6000ffe20c07.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2b6000ffe20c07.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2bf6e00075455d3c.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2bf6e00075455d3c.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2bfbf000c47880b7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2bfbf000c47880b7.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2c8c7000530eb0e7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2c8c7000530eb0e7.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2cdca0006185e7eb.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2cdca0006185e7eb.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2d2c6000adf6f6f4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2d2c6000adf6f6f4.jpg -------------------------------------------------------------------------------- /datasets/mydata/images/1066405,2d6f2000fec9dcab.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/images/1066405,2d6f2000fec9dcab.jpg -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,1b8000ef60354f.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,1b8000ef60354f.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2a72f000f214d26a.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2a72f000f214d26a.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2ac2400079a6d80f.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2ac2400079a6d80f.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2b6000ffe20c07.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2b6000ffe20c07.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2bf6e00075455d3c.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2bf6e00075455d3c.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2bfbf000c47880b7.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2bfbf000c47880b7.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2c8c7000530eb0e7.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2c8c7000530eb0e7.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2cdca0006185e7eb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2cdca0006185e7eb.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2d2c6000adf6f6f4.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2d2c6000adf6f6f4.txt -------------------------------------------------------------------------------- /datasets/mydata/labels/1066405,2d6f2000fec9dcab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/labels/1066405,2d6f2000fec9dcab.txt -------------------------------------------------------------------------------- /datasets/mydata/test.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/mydata/train.cache: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/train.cache -------------------------------------------------------------------------------- /datasets/mydata/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/train.txt -------------------------------------------------------------------------------- /datasets/mydata/val.cache: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/datasets/mydata/val.cache -------------------------------------------------------------------------------- /datasets/mydata/val.txt: -------------------------------------------------------------------------------- 1 | datasets/mydata/images/1066405,2bfbf000c47880b7.jpg 2 | -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/detect.py -------------------------------------------------------------------------------- /export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/export.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/common.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/__pycache__/common.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/experimental.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/__pycache__/experimental.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/yolo.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/__pycache__/yolo.cpython-37.pyc -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-bifpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-bifpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p34.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-p34.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5n6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5n6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-ghost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5s-ghost.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5s-transformer.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/tf.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5m_C3CA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_C3CA.yaml -------------------------------------------------------------------------------- /models/yolov5m_C3CBAM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_C3CBAM.yaml -------------------------------------------------------------------------------- /models/yolov5m_C3ECA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_C3ECA.yaml -------------------------------------------------------------------------------- /models/yolov5m_C3SE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_C3SE.yaml -------------------------------------------------------------------------------- /models/yolov5m_CBAM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_CBAM.yaml -------------------------------------------------------------------------------- /models/yolov5m_CoordAtt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_CoordAtt.yaml -------------------------------------------------------------------------------- /models/yolov5m_ECA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_ECA.yaml -------------------------------------------------------------------------------- /models/yolov5m_SE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5m_SE.yaml -------------------------------------------------------------------------------- /models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5n.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/setup.cfg -------------------------------------------------------------------------------- /split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/split.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/train.py -------------------------------------------------------------------------------- /tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/tutorial.ipynb -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/augmentations.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/augmentations.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/autoanchor.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/autoanchor.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/autobatch.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/autobatch.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/callbacks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/callbacks.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/datasets.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/datasets.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/downloads.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/downloads.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/general.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/general.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/loss.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/plots.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/plots.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/torch_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/__pycache__/torch_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/benchmarks.py -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/docker/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/docker/.dockerignore -------------------------------------------------------------------------------- /utils/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/docker/Dockerfile -------------------------------------------------------------------------------- /utils/docker/Dockerfile-cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/docker/Dockerfile-cpu -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/loggers/wandb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/README.md -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/loggers/wandb/__pycache__/wandb_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/__pycache__/wandb_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/loggers/wandb/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/log_dataset.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/sweep.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/sweep.yaml -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/val.py -------------------------------------------------------------------------------- /voc_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mohenghui/yolov5_attention/HEAD/voc_label.py --------------------------------------------------------------------------------