代码优化笔记
2018-10-10 本文已影响0人
壹刀_文
原代码:
X = []
for p in tqdm(train_fns):
img=cv2.imread(TRAIN_IMAGE_DIR+p,cv2.IMREAD_GRAYSCALE)
img = np.array(img,dtype=np.uint8)
X.append(img)
优化后的代码:
X =[np.array(cv2.imread(TRAIN_IMAGE_DIR+p,cv2.IMREAD_GRAYSCALE),dtype=np.uint8) for p in tqdm(train_fns)]