diff --git a/app/file_tool.py b/app/file_tool.py index c3eb34d..9e67b28 100644 --- a/app/file_tool.py +++ b/app/file_tool.py @@ -174,6 +174,7 @@ def mv_file(train_files: List[str], test_files: List[str], r_v_rate: Optional[fl def get_file(ori_path: str, type_list: Union[object,str]): # imgs = idc.get_image_all_proj_no(proj_no=proj_no, db=db) imgs = os.listdir(ori_path + '/images') + print('图像:',imgs) train_files = [] test_files = [] # 训练、测试比例强制9:1