├── .DS_Store ├── .idea ├── YOLOv5_with_BiFPN.iml ├── misc.xml ├── modules.xml ├── vcs.xml └── workspace.xml ├── README.md ├── data ├── coco128.yaml ├── hyp.finetune.yaml ├── hyp.scratch.yaml └── scripts │ └── get_coco.sh ├── detect.py ├── hubconf.py ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── common.cpython-37.pyc │ └── experimental.cpython-37.pyc ├── common.py ├── experimental.py ├── export.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5s-transformer.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── yolo.py └── yolov5x.yaml ├── requirements.txt ├── test.py ├── train.py ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── autoanchor.cpython-37.pyc │ ├── datasets.cpython-37.pyc │ ├── general.cpython-37.pyc │ ├── google_utils.cpython-37.pyc │ ├── metrics.cpython-37.pyc │ ├── plots.cpython-37.pyc │ └── torch_utils.cpython-37.pyc ├── activations.py ├── autoanchor.py ├── aws │ ├── __init__.py │ ├── mime.sh │ ├── resume.py │ └── userdata.sh ├── datasets.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── google_utils.py ├── loss.py ├── metrics.py ├── plots.py ├── torch_utils.py └── wandb_logging │ ├── __init__.py │ ├── log_dataset.py │ └── wandb_utils.py └── weights └── download_weights.sh /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.DS_Store -------------------------------------------------------------------------------- /.idea/YOLOv5_with_BiFPN.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.idea/YOLOv5_with_BiFPN.iml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /.idea/workspace.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/.idea/workspace.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/README.md -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/hyp.finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/data/hyp.finetune.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/data/hyp.scratch.yaml -------------------------------------------------------------------------------- /data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/detect.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/common.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/__pycache__/common.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/experimental.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/__pycache__/experimental.cpython-37.pyc -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/export.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s-transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5s-transformer.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/autoanchor.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/autoanchor.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/datasets.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/datasets.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/general.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/general.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/google_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/google_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/plots.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/plots.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/torch_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/__pycache__/torch_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/utils/wandb_logging/wandb_utils.py -------------------------------------------------------------------------------- /weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XingZeng307/YOLOv5_with_BiFPN/HEAD/weights/download_weights.sh --------------------------------------------------------------------------------