import os
import random # 后续用来将数据随机打乱和生成确定随机种子,保证每次生成的随机数据一样便于测试模型精准度
def _get_img_info(rng_seed,split_n,mode):
image_path_list = [] #用来存放图片的路径
label_path_list = [] #用来存放图片对应的标签
data_dir = r'C:UsersfutiangeDesktopZero to Heroexpression_test
aw_data'
for root,dirs,files in os.walk(data_dir):
for file in files:
path_file = os.path.join(root,file)
print(path_file)
if path_file.endswith(".jpg"): #判断该路径下文件是不是以.jpg结尾
#print(os.path.basename(root)) #输出图片路径
#print(os.path.basename(root)[0]) #输出该图片所在的文件夹的第一个字符,我这里文件夹的第一个字符就是图片的标签,测试时可以根据自己的文件夹名称更改
#print(int(os.path.basename(root)[0]))
image_path_list.append(path_file) #将图片路径加入列表
label_path_list.append(os.path.basename(root)[0]) #根据文件夹名称确定标签,并加入列表
data_info = [[n,l] for n,l in zip(image_path_list,label_path_list)] #将图片路径-标签 关联起来
random.seed(rng_seed) # 该方法中传入参数,确保每次生成的种子都是一样的
random.shuffle(data_info) #上一行代码生成的种子是确定的,保证了每次将列表元素打乱后的结果一样,便于测试模型性能
split_idx = int(len(data_info) * split_n) # data_len * 0.9 # split_n代表数据集划分的比例
if mode == 'train':
img_set = data_info[:split_idx]
elif mode == 'val':
img_set = data_info[split_idx:]
else:
raise Exception("mode 无法识别,仅支持(train,valid)")
return img_set #返回随机打乱后的数据集,后续在对其进行格式化即可将数据集加载进模型测试
if __name__ == '__main__':
_get_img_info(1,0.9,'train')
组织夹是什么详解python os.walk()方法的使用
未经允许不得转载:活检穿刺产品网 » 组织夹是什么详解python os.walk()方法的使用
相关推荐
动物拍片腹部怎么摆【译文推荐】小动物腹部x线摄影术,一文教会你如何拍摄出一张高质量的x光片!
动物拍片腹部怎么摆【病例分享】犬-外科-股骨头切除术
动物拍片腹部怎么摆【影像系列文章连载1】快速入门胸部X线片分析:总论
动物拍片腹部怎么摆《黑天鹅》导演达伦·阿伦诺夫斯基分享拍片工作方式
动物拍片腹部怎么摆伊沙,原名吴文健,文学家,以诗名世。1966年生于四川成都,1989年毕业于北京师范大学中文系,毕业后于西安外国语大学任教至今。现已出版著、译、编作品一百余部。获国内外数十项诗歌奖及其他文学类奖项。应邀出席国内外众多诗歌节、文学节和其他交流活动。代表性诗集有《车过黄河》《鸽子》《蓝灯》《无题》《唐
动物拍片腹部怎么摆11省发文,非必要不采购进口,国产优先!
动物怎么麻醉(W)动物麻醉机
动物怎么麻醉麻醉时,毛孩子是怎么“睡着”又“醒来”的?