From 9b008eaa99065555218f423e1b7a0d3f9d991a20 Mon Sep 17 00:00:00 2001
From: Zhang Wei <3160665918@qq.com>
Date: Thu, 5 Feb 2026 16:04:01 +0800
Subject: [PATCH] try_detect

---
 detect.sh                  |  3 +++
 my_gra_verify/verify_pt.py | 13 +++++++++++++
 2 files changed, 16 insertions(+)
 create mode 100644 detect.sh
 create mode 100644 my_gra_verify/verify_pt.py

diff --git a/detect.sh b/detect.sh
new file mode 100644
index 0000000..5e4a0d3
--- /dev/null
+++ b/detect.sh
@@ -0,0 +1,3 @@
+python /data6/zhangw/yolov5/detect.py \
+    --weights /data6/zhangw/ZW_gra_project/yolov5_result_pretrained/G0_cub100_yolov5s6_raw_e300/weights/best.rknn \
+    --source /data6/zhangw/ZW_gra_project/CUB_100/images/test/Acadian_Flycatcher_0073_795628.jpg
\ No newline at end of file
diff --git a/my_gra_verify/verify_pt.py b/my_gra_verify/verify_pt.py
new file mode 100644
index 0000000..239355c
--- /dev/null
+++ b/my_gra_verify/verify_pt.py
@@ -0,0 +1,13 @@
+import torch
+
+model = torch.load("/data6/zhangw/ZW_gra_project/yolov5_result_pretrained/G0_cub100_yolov5s6_raw_e300/weights/best.pt", map_location="cpu")["model"]
+model.eval()
+
+x = torch.randn(1, 3, 640, 640)
+y = model(x)
+
+if isinstance(y, (list, tuple)):
+    for o in y:
+        print(o.shape)
+else:
+    print(y.shape)