|
|
@@ -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",
|