Преглед изворни кода

不使用line_head,直接使用原始feature前5通道作为head

xue50 пре 8 месеци
родитељ
комит
4e244939b8
1 измењених фајлова са 1 додато и 2 уклоњено
  1. 1 2
      models/line_detect/train.yaml

+ 1 - 2
models/line_detect/train.yaml

@@ -1,7 +1,6 @@
 io:
   logdir: logs/
-  datadir: /root/LineDetect/models/line_detect/train_results/20250414_113846
-#  datadir: I:\datasets\wirenet_1000
+  datadir: /root/autodl-tmp/LineNet_Workpiece20250225
   resume_from:
   num_workers: 8
   tensorboard_port: 6000