From 4e8504abd9c1a7287dfcf9f96dfa04f061086cca Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 10 Sep 2022 13:25:01 +0300 Subject: [PATCH] PaddlePaddle Usage examples (#9358) --- 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 701b5b1ac..878cf48b6 100644 --- a/classify/predict.py +++ b/classify/predict.py @@ -22,6 +22,7 @@ Usage - formats: 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 bf808bc21..3c16ec809 100644 --- a/classify/val.py +++ b/classify/val.py @@ -17,6 +17,7 @@ Usage - formats: 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 69a1bf13a..a69606a3d 100644 --- a/detect.py +++ b/detect.py @@ -22,6 +22,7 @@ Usage - formats: 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 262b11a1a..9d33024a9 100644 --- a/export.py +++ b/export.py @@ -35,6 +35,7 @@ Inference: 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 396b5de0b..c601aacc8 100644 --- a/models/common.py +++ b/models/common.py @@ -312,7 +312,7 @@ class DetectMultiBackend(nn.Module): # 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 fed5e2157..4b0bdddae 100644 --- a/val.py +++ b/val.py @@ -16,6 +16,7 @@ Usage - formats: yolov5s.pb # TensorFlow GraphDef yolov5s.tflite # TensorFlow Lite yolov5s_edgetpu.tflite # TensorFlow Edge TPU + yolov5s_paddle_model # PaddlePaddle """ import argparse