【发布时间】:2019-07-11 01:53:06
【问题描述】:
我正在使用包含 3000 多张图像的数据集进行迁移学习。 这是代码的一部分:
import glob
import numpy as np
import os
import shutil
np.random.seed(42)
files = glob.glob('train/*')
cat_files = [fn for fn in files if 'cat' in fn]
dog_files = [fn for fn in files if 'dog' in fn]
len(cat_files), len(dog_files)
cat_train = np.random.choice(cat_files, size=1500, replace=False)
【问题讨论】:
标签: python-3.x deep-learning transfer-learning