From 495e685a3317319dea776ed6ebe2ca75df4ded3a Mon Sep 17 00:00:00 2001 From: glennjocher Date: Sat, 10 Sep 2022 13:08:24 +0300 Subject: [PATCH] PaddlePaddle Usage examples --- classify/predict.py | 1 + classify/val.py | 1 + detect.py | 1 + export.py | 1 + models/common.py | 2 +- val.py | 1 + 6 files changed, 6 insertions(+), 1 deletion(-) diff --git a/classify/predict.py b/classify/predict.py index 701b5b1ac92d..878cf48b6fef 100644 --- a/classify/predict.py +++ b/classify/predict.py @@ -22,6 +22,7 @@ yolov5s-cls.pb # TensorFlow GraphDef yolov5s-cls.tflite # TensorFlow Lite yolov5s-cls_edgetpu.tflite # TensorFlow Edge TPU + yolov5s-cls_paddle_model # PaddlePaddle """ import argparse diff --git a/classify/val.py b/classify/val.py index bf808bc21a84..3c16ec8092d8 100644 --- a/classify/val.py +++ b/classify/val.py @@ -17,6 +17,7 @@ yolov5s-cls.pb # TensorFlow GraphDef yolov5s-cls.tflite # TensorFlow Lite yolov5s-cls_edgetpu.tflite # TensorFlow Edge TPU + yolov5s-cls_paddle_model # PaddlePaddle """ import argparse diff --git a/detect.py b/detect.py index 69a1bf13aac6..a69606a3dff9 100644 --- a/detect.py +++ b/detect.py @@ -22,6 +22,7 @@ yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU + yolov5s_paddle_model # PaddlePaddle """ import argparse diff --git a/export.py b/export.py index 262b11a1a268..9d33024a9ca4 100644 --- a/export.py +++ b/export.py @@ -35,6 +35,7 @@ yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU + yolov5s_paddle_model # PaddlePaddle TensorFlow.js: $ cd .. && git clone https://github.com/zldrobit/tfjs-yolov5-example.git && cd tfjs-yolov5-example diff --git a/models/common.py b/models/common.py index 396b5de0b505..c601aacc885c 100644 --- a/models/common.py +++ b/models/common.py @@ -312,7 +312,7 @@ def __init__(self, weights='yolov5s.pt', device=torch.device('cpu'), dnn=False, # PyTorch: weights = *.pt # TorchScript: *.torchscript # ONNX Runtime: *.onnx - # ONNX OpenCV DNN: *.onnx with --dnn + # ONNX OpenCV DNN: *.onnx --dnn # OpenVINO: *.xml # CoreML: *.mlmodel # TensorRT: *.engine diff --git a/val.py b/val.py index fed5e21577e5..4b0bdddae3b1 100644 --- a/val.py +++ b/val.py @@ -16,6 +16,7 @@ yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU + yolov5s_paddle_model # PaddlePaddle """ import argparse