用的pytorch來訓練deeplabv3+

在做deeplabv3+的過程中，我的訓練圖片是8位的，如下圖：

8位的：

24位的：

這樣雖然在訓練過程中能夠正常訓練。但是在評估過程中會出錯，所以決定將訓練圖片轉成24位圖，重新訓練。最后結果也表明了，只要將訓練圖片轉成24位后之后的評估可視化等都沒有問題。

由于RGB的圖片就為24位，則簡單將圖片利用PIL轉為RGB格式即可

#!/usr/bin/env python2# -*- coding: utf-8 -*-"""Created on Tue Oct 24 10:47:36 2018@author: yxh"""import numpy as npfrom PIL import Imageimport matplotlib.pyplot as pltimport osimport sysimport shutilpath='/home/yxh/caffe/examples/fcn/IMAGES/IMAGES/'newpath='/home/yxh/caffe/examples/fcn/IMAGES/output/'def turnto24(path): fileList = [] files = os.listdir(path) i=0 for f in files:  imgpath = path + '/' +f  img=Image.open(f).convert('RGB')  dirpath = newpath   file_name, file_extend = os.path.splitext(f)  dst = os.path.join(os.path.abspath(dirpath), file_name + '.jpg')  img.save(dst)turnto24(path)

總結

以上所述是小編給大家介紹的Python將8位的圖片轉為24位的圖片，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對武林站長站網站的支持！