Skip to content

Commit 76135a9

Browse files
committed
* 去除不编译的文件,包括其import的环境也不写入
1 parent 4cb3273 commit 76135a9

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,5 @@
33

44
```bash
55
python main.py --dataset_type paddle_detection --input_dataset_dir F:\数据集\VOC数据集\箱门检测数据集\ContainVOC --save_dataset_dir E:\Data\VOC数据集\箱门检测数据集\ContainVOC
6+
python main.py --dataset_type paddle_text_detection --input_dataset_dir F:\数据集\关键点检测数据集\箱号关键点数据集 --save_dataset_dir E:\Data\关键点检测数据集\箱号关键点数据集
67
```

main.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
def test_create_paddle_years_datasets(args):
1414
CreateYearsDatasets(args.input_dataset_dir,None,save_path=args.save_dataset_dir,rate=0.9)
1515

16+
1617
if __name__ == '__main__':
1718
import argparse
1819
parser = argparse.ArgumentParser(description="制作数据集脚本")
@@ -26,6 +27,8 @@ def test_create_paddle_years_datasets(args):
2627
CreateYearsDatasets(args.input_dataset_dir, None, save_path=args.save_dataset_dir, rate=0.9)
2728
elif args.dataset_type == "yolo_detection":
2829
CreateDarknetVocDatasets(args.input_dataset_dir, save_path=args.save_dataset_dir, rate=0.9, VOC_CLASSES=args.voc_labels)
30+
elif args.dataset_type == "paddle_text_detection":
31+
create_text_detection_datasets(args.input_dataset_dir, args.save_dataset_dir, split_rate=0.95)
2932

3033
#removeNolabelDatasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集\2022-03-09")
3134
#create_text_detection_datasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集",r"E:\Data\字符检测识别数据集\定制版箱号关键点数据集",0.95)

0 commit comments

Comments
 (0)