1. 問題描述
- 將博客https://blog.csdn.net/ShadowN1ght/article/details/78571187的代碼運行與不同訓練集時報以下錯誤:
“if int(img.getpixel((w, h))) < 190:
**if int(img.getpixel((w, h))) < 190:
TypeError: int() argument must be a string, a bytes-like object or a number, not ‘tuple’
**
” - 在使用PIL.Image的getpixel()方法時,因爲輸入圖片不同返回的是tuple 或者int類型的數據
2. 原因分析
- getpixel()函數在輸入圖片時三通道的時候,輸出時一個 'tuple’格式,在輸入圖片時單通道的時候,輸出時一個 'int’格式
- 原博客使用的數據集是單通道,而我的訓練集是三通道
3. 處理方法
- 將輸入的圖片轉換爲單通道即可
- 實現的代碼
img = Image.open(path)
img = img.convert("I") # 將三通道變成了單通道