فهرست منبع

feat: 修改模型名称为SLANet_plus,注释掉旧模型配置

zhch158_admin 1 ماه پیش
والد
کامیت
acd9981cf8
1فایلهای تغییر یافته به همراه9 افزوده شده و 1 حذف شده
  1. 9 1
      zhch/model_single_process.py

+ 9 - 1
zhch/model_single_process.py

@@ -242,7 +242,15 @@ if __name__ == "__main__":
         #     "--device": "cpu",
         # }
         
-        model_name = "RT-DETR-L_wireless_table_cell_det"
+        # model_name = "RT-DETR-L_wireless_table_cell_det"
+        # demo = {
+        #     "--model_name": model_name,
+        #     "--input_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.img",
+        #     "--output_dir": f"/Users/zhch158/workspace/data/流水分析/B用户_扫描流水/{model_name}_Results",
+        #     "--device": "cpu",
+        # }
+
+        model_name = "SLANet_plus"
         demo = {
             "--model_name": model_name,
             "--input_dir": "/Users/zhch158/workspace/data/流水分析/B用户_扫描流水.img",