Quellcode durchsuchen

[feat] 修改参数名

WangChao vor 2 Monaten
Ursprung
Commit
298cc15572
1 geänderte Dateien mit 4 neuen und 4 gelöschten Zeilen
  1. 4 4
      data_collection/clean_label_la_0.1.4l.py

+ 4 - 4
data_collection/clean_label_la_0.1.4l.py

@@ -72,11 +72,11 @@ def clean_label_files(labels_dir, output_labels_dir, log_file):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser("删除特定标注信息\n")
-    parser.add_argument("image_dir", help="图片文件夹路径")
+    parser.add_argument("dataset_dir", help="图片文件夹路径")
     parser.add_argument("--log_file", default="processed_files.log", help="记录被处理文件的日志文件路径")
 
     args = parser.parse_args()
 
-    output_labels_dir = os.path.join(args.image_dir, 'processed_labels')  # 生成的新的labels文件夹
-    log_file_dir = os.path.join(args.image_dir, args.log_file)  # 日志文件
-    clean_label_files(os.path.join(args.image_dir, 'labels'), output_labels_dir, log_file_dir)
+    output_labels_dir = os.path.join(args.dataset_dir, 'processed_labels')  # 生成的新的labels文件夹
+    log_file_dir = os.path.join(args.dataset_dir, args.log_file)  # 日志文件
+    clean_label_files(os.path.join(args.dataset_dir, 'labels'), output_labels_dir, log_file_dir)