PaddlePaddle Usage examples (#9358)

ultralytics · Sep 10, 2022 · 4e8504a · 4e8504a
1 parent e3e5122
commit 4e8504a
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -16,6 +16,7 @@
                               yolov5s.pb                 # TensorFlow GraphDef
                               yolov5s.tflite             # TensorFlow Lite
                               yolov5s_edgetpu.tflite     # TensorFlow Edge TPU
+                              yolov5s_paddle_model       # PaddlePaddle
 """
 
 import argparse