From 89e38dc645324e7fe30062f5e65283e9ca3dcdc3 Mon Sep 17 00:00:00 2001 From: nihui Date: Thu, 7 Jul 2022 19:23:03 +0800 Subject: [PATCH] Update README.md --- README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/README.md b/README.md index 483c7c1..cddedaa 100644 --- a/README.md +++ b/README.md @@ -181,6 +181,11 @@ TRAIN: ``` python3 test.py --yaml configs/coco.yaml --weight weights/coco_ap05_0.250_280epoch.pth --img data/3.jpg --onnx ``` +## Export torchscript +* You can export .pt by adding the --torchscript option when executing test.py + ``` + python3 test.py --yaml configs/coco.yaml --weight weights/coco_ap05_0.250_280epoch.pth --img data/3.jpg --torchscript + ``` ## onnx-runtime * You can learn about the pre and post-processing methods of FastestDet in this Sample ```