https://github.com/thnak/yolov7-2-tensorflow
export any your YOLOv7 model to TensorFlow, TensorFlowJs, ONNX, OpenVINO, RKNN,...
https://github.com/thnak/yolov7-2-tensorflow
deep-learning directml object-detection rknn tensorflow tensorflowjs tfjs windows yolov7
Last synced: 5 months ago
JSON representation
export any your YOLOv7 model to TensorFlow, TensorFlowJs, ONNX, OpenVINO, RKNN,...
- Host: GitHub
- URL: https://github.com/thnak/yolov7-2-tensorflow
- Owner: thnak
- License: gpl-3.0
- Created: 2022-11-24T17:47:36.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2025-02-07T09:06:54.000Z (over 1 year ago)
- Last Synced: 2025-02-07T10:20:36.567Z (over 1 year ago)
- Topics: deep-learning, directml, object-detection, rknn, tensorflow, tensorflowjs, tfjs, windows, yolov7
- Language: Python
- Homepage:
- Size: 612 MB
- Stars: 14
- Watchers: 0
- Forks: 4
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE.md
Awesome Lists containing this project
README
YOLOv7 supported for tfjs, tflite, saved model, openvino, onnx, rknn.Support for WINDOWS and LINUX
Train
!python train.py --batch 4 --epochs 10 --data 'data/coco.yaml' --cache_images disk --cfg 'cfg/yolov7-tiny.yaml' --device dml:0 --imgsz 640
dml:0 to set dml device (windows), cuda:0 to set cuda device, cache-images for faster training disk/ram/no
Export
python export.py --weights "yolov7.pt" --include tfjs/onnx/saved_model/openvino/tflite
for more information
python export.py --help
Usage
tfjs model can be used with makesense.ai
🔥Actions -> Run AI locally -> YOLOv5 object detection using rectangles -> and import all files from *_web_model folder
also you can follow this step to use (from https://github.com/Hyuto/yolov5-tfjs)
.onnx can be used with python.exe detect.py --weights "yolov7-w6.onnx" --nosave --source 0 --view-img 1 (use 0 to freeze screen)
References