machine learning/deep learning

代码优化笔记

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)]

上一篇 下一篇

猜你喜欢

热点阅读