├── README.md ├── __pycache__ └── py_quant_utils.cpython-37.pyc ├── py_quant_utils.py ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── augmentations.cpython-37.pyc │ ├── autoanchor.cpython-37.pyc │ ├── callbacks.cpython-37.pyc │ ├── dataloaders.cpython-37.pyc │ ├── downloads.cpython-37.pyc │ ├── general.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 ├── dataloaders.py ├── docker │ ├── Dockerfile │ ├── Dockerfile-arm64 │ └── 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 │ ├── clearml │ │ ├── README.md │ │ ├── __init__.py │ │ ├── clearml_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 └── torch_utils.py ├── weights ├── confusion_matrix.png ├── val_batch0_labels.jpg ├── val_batch1_labels.jpg ├── val_batch2_labels.jpg ├── yolov8n-max-1024.pth ├── yolov8n-max-2048.pth └── yolov8n_qat_detect_3.onnx ├── yolov8_flow_qat_int8.py ├── yolov8_ptq_int8.py └── yolov8_ptq_sensitive_int8.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/py_quant_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/__pycache__/py_quant_utils.cpython-37.pyc -------------------------------------------------------------------------------- /py_quant_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/py_quant_utils.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/augmentations.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/augmentations.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/autoanchor.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/autoanchor.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/callbacks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/callbacks.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dataloaders.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/dataloaders.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/downloads.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/downloads.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/general.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/general.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/plots.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/plots.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/torch_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/__pycache__/torch_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/aws/mime.sh -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/aws/userdata.sh -------------------------------------------------------------------------------- /utils/benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/benchmarks.py -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/dataloaders.py -------------------------------------------------------------------------------- /utils/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/docker/Dockerfile -------------------------------------------------------------------------------- /utils/docker/Dockerfile-arm64: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/docker/Dockerfile-arm64 -------------------------------------------------------------------------------- /utils/docker/Dockerfile-cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/docker/Dockerfile-cpu -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/clearml/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/clearml/README.md -------------------------------------------------------------------------------- /utils/loggers/clearml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/clearml/clearml_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/clearml/clearml_utils.py -------------------------------------------------------------------------------- /utils/loggers/clearml/hpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/clearml/hpo.py -------------------------------------------------------------------------------- /utils/loggers/wandb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/wandb/README.md -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/wandb/log_dataset.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/wandb/sweep.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/wandb/sweep.yaml -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /weights/confusion_matrix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/confusion_matrix.png -------------------------------------------------------------------------------- /weights/val_batch0_labels.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/val_batch0_labels.jpg -------------------------------------------------------------------------------- /weights/val_batch1_labels.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/val_batch1_labels.jpg -------------------------------------------------------------------------------- /weights/val_batch2_labels.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/val_batch2_labels.jpg -------------------------------------------------------------------------------- /weights/yolov8n-max-1024.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/yolov8n-max-1024.pth -------------------------------------------------------------------------------- /weights/yolov8n-max-2048.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/yolov8n-max-2048.pth -------------------------------------------------------------------------------- /weights/yolov8n_qat_detect_3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/weights/yolov8n_qat_detect_3.onnx -------------------------------------------------------------------------------- /yolov8_flow_qat_int8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/yolov8_flow_qat_int8.py -------------------------------------------------------------------------------- /yolov8_ptq_int8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/yolov8_ptq_int8.py -------------------------------------------------------------------------------- /yolov8_ptq_sensitive_int8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huangzongmou/yolov8-pytorch_quantization/HEAD/yolov8_ptq_sensitive_int8.py --------------------------------------------------------------------------------