Julia Flux,神经网络中具有不同维度的图像

时间:2019-03-11 10:24:25

标签: image neural-network julia conv-neural-network flux

我想知道细胞是否感染了疟疾(https://www.kaggle.com/iarunava/cell-images-for-detecting-malaria)。 但是图片的大小不同,因此我想将每个图像调整为相同的大小,以便能够使用我的第一层(具有静态大小)

如何在Julia中调整图像大小?还是可以仅使用流量来计算不同尺寸的图像?

1 个答案:

答案 0 :(得分:5)

您可以使用软件包ImagesImageMagick(您需要同时安装两者)来执行此操作,然后:

using Images
download("https://juliaimages.org/latest/assets/logo.png","test.png");
img = Images.load("test.png");
size(img) # outputs (128, 128)
img2 = imresize(img,(50,50));
Images.save("test2.png",img2);

文件test2.png的大小为50x50。有关imsize的更多信息,请访问:

https://juliaimages.org/latest/function_reference.html#Spatial-transformations-and-resizing-1

您应该事先进行此操作(即在Flux.jl之外),因为否则每次计算梯度时都需要进行插值。

软件包Interpolations也将允许您调整图像的大小。然后,您将分别对红色,绿色,蓝色通道进行插值。

对于kaggle挑战的特定图像,也可以选择用黑色像素填充数据到相同大小。