├── .gitignore ├── LICENSE ├── README.md ├── convert_yolo.py ├── convert_yolov5.py ├── convertor ├── __init__.py ├── yolo │ ├── convert_tf_onnx.py │ └── convert_tf_tflite.py └── yolov5 │ ├── __init__.py │ ├── convert_onnx_tf.py │ ├── convert_onnx_vino.py │ ├── convert_tf_onnx.py │ ├── convert_tf_tflite.py │ └── convert_torch_onnx.py ├── datasets ├── .gitignore ├── README.md ├── create_small_dataset.py └── download_coco_val2017.sh ├── detect.py ├── detect_all.sh ├── detector ├── .gitignore ├── __init__.py ├── base.py ├── yolo.py └── yolov5.py ├── download_font.sh ├── download_superres.sh ├── evaluate_all.sh ├── ipynb ├── accumulate.ipynb ├── map.png ├── time.png └── time_vs_map.png ├── labels └── coco_labels.txt ├── models ├── __init__.py ├── common.py ├── experimental.py ├── export.py ├── tf_yolo.py ├── tf_yolov5.py ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── requirements.txt ├── tools ├── convert_yolo_config_csv.py ├── yolov3-spp.cfg ├── yolov3-spp.csv ├── yolov3-spp.xlsx ├── yolov3-tiny.cfg ├── yolov3-tiny.csv ├── yolov3-tiny.xlsx ├── yolov3.cfg ├── yolov3.csv ├── yolov3.xlsx ├── yolov4-csp.cfg ├── yolov4-csp.csv ├── yolov4-csp.xlsx ├── yolov4-tiny.cfg ├── yolov4-tiny.csv ├── yolov4-tiny.xlsx ├── yolov4.cfg ├── yolov4.csv ├── yolov4.xlsx ├── yolov4x-mish.cfg ├── yolov4x-mish.csv └── yolov4x-mish.xlsx ├── utils ├── __init__.py ├── activations.py ├── autoanchor.py ├── convert_tflite.py ├── datasets.py ├── general.py ├── google_utils.py ├── image_preproc.py ├── loss.py ├── metrics.py ├── nms.py ├── plots.py └── torch_utils.py └── weights ├── .gitignore ├── Dockerfile ├── README.md ├── build_docker.sh ├── compile_edgetpu.sh ├── download_yolo.sh └── download_yolov5.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/README.md -------------------------------------------------------------------------------- /convert_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convert_yolo.py -------------------------------------------------------------------------------- /convert_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convert_yolov5.py -------------------------------------------------------------------------------- /convertor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /convertor/yolo/convert_tf_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolo/convert_tf_onnx.py -------------------------------------------------------------------------------- /convertor/yolo/convert_tf_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolo/convert_tf_tflite.py -------------------------------------------------------------------------------- /convertor/yolov5/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /convertor/yolov5/convert_onnx_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolov5/convert_onnx_tf.py -------------------------------------------------------------------------------- /convertor/yolov5/convert_onnx_vino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolov5/convert_onnx_vino.py -------------------------------------------------------------------------------- /convertor/yolov5/convert_tf_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolov5/convert_tf_onnx.py -------------------------------------------------------------------------------- /convertor/yolov5/convert_tf_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolov5/convert_tf_tflite.py -------------------------------------------------------------------------------- /convertor/yolov5/convert_torch_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/convertor/yolov5/convert_torch_onnx.py -------------------------------------------------------------------------------- /datasets/.gitignore: -------------------------------------------------------------------------------- 1 | annotations 2 | val2017 3 | sample_dataset 4 | -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/create_small_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/datasets/create_small_dataset.py -------------------------------------------------------------------------------- /datasets/download_coco_val2017.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/datasets/download_coco_val2017.sh -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/detect.py -------------------------------------------------------------------------------- /detect_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/detect_all.sh -------------------------------------------------------------------------------- /detector/.gitignore: -------------------------------------------------------------------------------- 1 | TakaoGothic.ttf 2 | -------------------------------------------------------------------------------- /detector/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detector/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/detector/base.py -------------------------------------------------------------------------------- /detector/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/detector/yolo.py -------------------------------------------------------------------------------- /detector/yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/detector/yolov5.py -------------------------------------------------------------------------------- /download_font.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/download_font.sh -------------------------------------------------------------------------------- /download_superres.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/download_superres.sh -------------------------------------------------------------------------------- /evaluate_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/evaluate_all.sh -------------------------------------------------------------------------------- /ipynb/accumulate.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/ipynb/accumulate.ipynb -------------------------------------------------------------------------------- /ipynb/map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/ipynb/map.png -------------------------------------------------------------------------------- /ipynb/time.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/ipynb/time.png -------------------------------------------------------------------------------- /ipynb/time_vs_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/ipynb/time_vs_map.png -------------------------------------------------------------------------------- /labels/coco_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/labels/coco_labels.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/common.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/export.py -------------------------------------------------------------------------------- /models/tf_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/tf_yolo.py -------------------------------------------------------------------------------- /models/tf_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/tf_yolov5.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/yolov5l.yaml -------------------------------------------------------------------------------- /models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/yolov5m.yaml -------------------------------------------------------------------------------- /models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/yolov5s.yaml -------------------------------------------------------------------------------- /models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/models/yolov5x.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/convert_yolo_config_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/convert_yolo_config_csv.py -------------------------------------------------------------------------------- /tools/yolov3-spp.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-spp.cfg -------------------------------------------------------------------------------- /tools/yolov3-spp.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-spp.csv -------------------------------------------------------------------------------- /tools/yolov3-spp.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-spp.xlsx -------------------------------------------------------------------------------- /tools/yolov3-tiny.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-tiny.cfg -------------------------------------------------------------------------------- /tools/yolov3-tiny.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-tiny.csv -------------------------------------------------------------------------------- /tools/yolov3-tiny.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3-tiny.xlsx -------------------------------------------------------------------------------- /tools/yolov3.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3.cfg -------------------------------------------------------------------------------- /tools/yolov3.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3.csv -------------------------------------------------------------------------------- /tools/yolov3.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov3.xlsx -------------------------------------------------------------------------------- /tools/yolov4-csp.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-csp.cfg -------------------------------------------------------------------------------- /tools/yolov4-csp.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-csp.csv -------------------------------------------------------------------------------- /tools/yolov4-csp.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-csp.xlsx -------------------------------------------------------------------------------- /tools/yolov4-tiny.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-tiny.cfg -------------------------------------------------------------------------------- /tools/yolov4-tiny.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-tiny.csv -------------------------------------------------------------------------------- /tools/yolov4-tiny.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4-tiny.xlsx -------------------------------------------------------------------------------- /tools/yolov4.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4.cfg -------------------------------------------------------------------------------- /tools/yolov4.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4.csv -------------------------------------------------------------------------------- /tools/yolov4.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4.xlsx -------------------------------------------------------------------------------- /tools/yolov4x-mish.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4x-mish.cfg -------------------------------------------------------------------------------- /tools/yolov4x-mish.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4x-mish.csv -------------------------------------------------------------------------------- /tools/yolov4x-mish.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/tools/yolov4x-mish.xlsx -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/convert_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/convert_tflite.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/google_utils.py -------------------------------------------------------------------------------- /utils/image_preproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/image_preproc.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/nms.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /weights/.gitignore: -------------------------------------------------------------------------------- 1 | yolo 2 | yolov5 3 | -------------------------------------------------------------------------------- /weights/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/weights/Dockerfile -------------------------------------------------------------------------------- /weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/weights/README.md -------------------------------------------------------------------------------- /weights/build_docker.sh: -------------------------------------------------------------------------------- 1 | #!/bin/sh 2 | docker build -t edgetpu_env . 3 | -------------------------------------------------------------------------------- /weights/compile_edgetpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/weights/compile_edgetpu.sh -------------------------------------------------------------------------------- /weights/download_yolo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/weights/download_yolo.sh -------------------------------------------------------------------------------- /weights/download_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tetutaro/yolo_various_framework/HEAD/weights/download_yolov5.py --------------------------------------------------------------------------------