├── .gitignore ├── LICENSE ├── README.md ├── configs ├── accuracy-check-scaled-yolov4-p5.yml ├── accuracy-check-scaled-yolov4-p6.yml ├── accuracy-check-scaled-yolov4-p7.yml ├── accuracy-check-yolov3.yml ├── accuracy-check-yolov4.yml └── accuracy-check-yolov5.yml ├── data ├── coco.yaml ├── coco128.yaml ├── hyp.finetune.yaml ├── hyp.scratch.yaml ├── images │ ├── bus.jpg │ └── zidane.jpg ├── scripts │ ├── get_coco.sh │ └── get_voc.sh └── voc.yaml ├── demo_result.png ├── detect.py ├── images ├── bus.jpg └── zidane.jpg ├── models ├── __init__.py ├── common.py ├── experimental.py ├── export.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov4-csp.yaml │ ├── yolov4-p5.yaml │ ├── yolov4-p6.yaml │ ├── yolov4-p7.yaml │ ├── yolov4-pacsp-mish.yaml │ ├── yolov4-pacsp-s-mish.yaml │ ├── yolov4-pacsp-x-mish.yaml │ ├── yolov4-tiny-mish.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l.yaml │ ├── yolov5m.yaml │ ├── yolov5s.yaml │ └── yolov5x.yaml └── yolo.py ├── requirements.txt ├── test.py ├── tools └── accuracy_checker │ └── accuracy_checker │ ├── adapters │ ├── __init__.py │ └── yolo.py │ ├── postprocessor │ └── nms.py │ └── topology_types.py ├── train.py ├── utils ├── __init__.py ├── activations.py ├── autoanchor.py ├── datasets.py ├── general.py ├── google_utils.py ├── loss.py ├── metrics.py ├── plots.py └── torch_utils.py ├── weights └── download_weights.sh ├── yolo_openvino_demo.py └── yolov5s_output_node.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/README.md -------------------------------------------------------------------------------- /configs/accuracy-check-scaled-yolov4-p5.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-scaled-yolov4-p5.yml -------------------------------------------------------------------------------- /configs/accuracy-check-scaled-yolov4-p6.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-scaled-yolov4-p6.yml -------------------------------------------------------------------------------- /configs/accuracy-check-scaled-yolov4-p7.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-scaled-yolov4-p7.yml -------------------------------------------------------------------------------- /configs/accuracy-check-yolov3.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-yolov3.yml -------------------------------------------------------------------------------- /configs/accuracy-check-yolov4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-yolov4.yml -------------------------------------------------------------------------------- /configs/accuracy-check-yolov5.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/configs/accuracy-check-yolov5.yml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/coco128.yaml -------------------------------------------------------------------------------- /data/hyp.finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/hyp.finetune.yaml -------------------------------------------------------------------------------- /data/hyp.scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/hyp.scratch.yaml -------------------------------------------------------------------------------- /data/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/images/bus.jpg -------------------------------------------------------------------------------- /data/images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/images/zidane.jpg -------------------------------------------------------------------------------- /data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /data/scripts/get_voc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/scripts/get_voc.sh -------------------------------------------------------------------------------- /data/voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/data/voc.yaml -------------------------------------------------------------------------------- /demo_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/demo_result.png -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/detect.py -------------------------------------------------------------------------------- /images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/images/bus.jpg -------------------------------------------------------------------------------- /images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/images/zidane.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/export.py -------------------------------------------------------------------------------- /models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/anchors.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-csp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-csp.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-p5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-p5.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-pacsp-mish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-pacsp-mish.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-pacsp-s-mish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-pacsp-s-mish.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-pacsp-x-mish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-pacsp-x-mish.yaml -------------------------------------------------------------------------------- /models/hub/yolov4-tiny-mish.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov4-tiny-mish.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /models/hub/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5l.yaml -------------------------------------------------------------------------------- /models/hub/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5m.yaml -------------------------------------------------------------------------------- /models/hub/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5s.yaml -------------------------------------------------------------------------------- /models/hub/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/hub/yolov5x.yaml -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/models/yolo.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/test.py -------------------------------------------------------------------------------- /tools/accuracy_checker/accuracy_checker/adapters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/tools/accuracy_checker/accuracy_checker/adapters/__init__.py -------------------------------------------------------------------------------- /tools/accuracy_checker/accuracy_checker/adapters/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/tools/accuracy_checker/accuracy_checker/adapters/yolo.py -------------------------------------------------------------------------------- /tools/accuracy_checker/accuracy_checker/postprocessor/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/tools/accuracy_checker/accuracy_checker/postprocessor/nms.py -------------------------------------------------------------------------------- /tools/accuracy_checker/accuracy_checker/topology_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/tools/accuracy_checker/accuracy_checker/topology_types.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/weights/download_weights.sh -------------------------------------------------------------------------------- /yolo_openvino_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/yolo_openvino_demo.py -------------------------------------------------------------------------------- /yolov5s_output_node.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chen-MingChang/pytorch_YOLO_OpenVINO_demo/HEAD/yolov5s_output_node.png --------------------------------------------------------------------------------