摘要: import random import numpy as np import pandas as pd import cv2 def date_loader(image_dir, file_name, batch_size=1, mode='train'): train_dir_list = [] 阅读全文
posted @ 2021-08-06 17:52 沐沐子枫 阅读(157) 评论(0) 推荐(1) 编辑
摘要: #切分数据集 img_dir = train_parameters['img_dir'] file_name = train_parameters['file_name'] df = pd.read_csv(file_name) df = df.sample(frac=1) train_list = 阅读全文
posted @ 2021-08-06 17:51 沐沐子枫 阅读(110) 评论(0) 推荐(0) 编辑