问题设置
我有一个包含N
张图片的数据集。
必须在此数据集上从头开始训练某个网络(例如Alexnet)。
对于每张图片,都会生成10
个增强版本。这些增强涉及调整大小,裁剪和翻转。例如,必须使用最小尺寸为256
像素来调整图像大小,然后对其进行224 x 224
的随机裁剪。然后它必须翻转。 5
必须采取这种随机作物,并且还必须准备翻转的版本。
这些增强版本必须进入网络进行培训而不是原始图片
另外非常有益的是,数据集中的多个图像并行扩充并放入队列或任何容器中,batchsize
个样本被推入GPU进行训练
上下文
我的理解
就我的搜索而言,我在CNTK中找不到任何可以做到这一点的框架。
问题
答案 0 :(得分:0)
请查看CNTK 201教程:
图像阅读器具有内置的转换功能,可满足您的许多要求。不幸的是,它不在GPU中。