tensorflow裏面聲明圖片的形狀

tensorflow有時候要對圖片進行裁剪拼接等操作,基本上內置函數在裁剪的時候都會隱形的返回圖片形狀的tensor,但是有時候默寫函數不會返回圖片的size,所以讓我們在後面抓取圖片的形狀信息的時候會很麻煩,於是有必要在op不返回size的時候,聲明size的信息。

# Set static shape so that tensorflow knows shape at compile time. 
img_crop.set_shape((crop_h, crop_w, 4))
label_crop.set_shape((crop_h,crop_w, 1))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章