|
@@ -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)
|