做昆蟲數據集圖片分類的時候,將圖片用opencv讀入,存入numpy的時候出現了以下的錯誤:
error: (-215:Assertion failed) !ssize.empty() in function 'cv::resize’
找了好久沒發現問題,以為是圖片的size搞錯了,反復檢查幾遍沒有錯誤。后面上網搜了一下,可能是有的圖片是損壞的,或者是圖片的路徑輸入問題,或者是size問題。
后面嘗試后發現是有None的圖片(學長給數據集的時候忘記把腳本刪除了qwq)。
cv2讀入的時候一直是None,導致錯誤。
用以下代碼找出錯誤圖片的名稱:
1 for i, file in enumerate(image_dir): 2 # print(file) 3 img = cv2.imread(os.path.join("image", file)) 4 if img is None: 5 print(os.path.join("image", file)) 6 break