[问题] numpy维度问题

楼主: grimaceo (哈囉哈囉)   2020-05-16 11:17:41
各位前辈好 小的最近在建构CNN时遇到一些问题
当我读进训练集N张照片时train.shape会显示(N,1) train[0][0].shape就会显示(height
,w
但model要读资料时会要求资料要是四维的array,网络上好像都找不到能把二维转成四维

试过resize但资料好像会跑掉 想请问各位:
有什么方法可以把二维array中的array降维变成4D矩阵,或是有什么方法让CNN网络能够
读?
谢谢各位
作者: littlebo1990 (Bo)   2020-05-16 11:54:00
train.reshap(-1,3,3,1) 选我正解抱歉搞错了XD 我以为你w,h分别为3, channel是1..应该是你的np array里面是object, 所以shape才是你可以用 list comprehension 把里面的object变arrai.e train = np.array([i.reshape(w,h,3) for i intrain])这样train dim就会变(N,w,h,3)
作者: eooft   2020-05-16 13:11:00
那个3代表channel数 先弄清楚你的channel是1(黑白)还3(RGB)

Links booklink

Contact Us: admin [ a t ] ucptt.com