[問題] mask rcnn訓練自己的資料集已刪文
在上上週開始學ML
上週試著用colab跑過TORCHVISION OBJECT DETECTION FINETUNING TUTORIAL
這週被指定的作業是要將PennFudanPed取代成自己的資料集
下午試做之後發現在pos=np.where(masks[i])出現錯誤
說TypeError: 'bool' object is not subscriptable
我猜有可能是因為
PennFudanPed辨別的只有人類 但我們的資料集有辨別超過一種東西
而且PennFudanPed給的mask光看圖片是全黑的 需要自己用mask.putpalette填色
我們的資料集給的mask上面已經有按照分類跟bounding box各自填色了
後來問前輩
前輩說可以把我們mask裡面的多種顏色全部改成白色 就改成1的形式
但我沒有聽得很懂qq想了一個下午也不太知道該怎麼做
有試過用convert(‘1’)或convert(‘L’)似乎也不對
麻煩版友賜教 謝謝
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 110.28.33.185 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1627299629.A.90A.html
※ 編輯: sjsjsjlove (110.28.33.185 臺灣), 07/26/2021 19:42:20
DataScience 近期熱門文章
PTT數位生活區 即時熱門文章
