摘要:
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 = [] 阅读全文
摘要:
#切分数据集 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 = 阅读全文