def get_train_frame_dims():
img_path = glob(os.path.join(TRAIN_DIR, '*/*'))[0]
img = imread(img_path, mode='RGB')
shape = np.shape(img)
return shape[0], shape[1]
微信
公众号
扫码关注公众号