為什麼輸入generator要是四維
2019/12/18 上午 09:03
電腦視覺深度學習討論版
林辰
觀看數：14
回答數：1
收藏數：0
cvdl-1
cvdl-1-d16
##輸入generator要是四維，(224,224,3)變成(4,224,224,3)
可以改成6.7.8....維讓她一次多生成一些變化嗎?
回答列表
-
2019/12/18 上午 11:00楊哲寧贊同數：0不贊同數：0留言數：0
您好，這裡4維的意思為(batch_size,height,width,channel)，所以我們當然可以變動batch_size的大小，如(5,224,224,3)，但不能新增維度。
batch_gen本身就可以一直產出新的隨機組合，所以程式碼下方才有一個break，不然他就會變成一個endless loop，kernel就會當掉。