深度學(xué)習(xí)之用python計(jì)算數(shù)據(jù)集均值

import?os??

from?PIL?import?Image????

import?matplotlib.pyplot?as?plt??

import?numpy?as?np??

from?scipy.misc?import?imread???


filepath?='/home/JPEGImages'?#?數(shù)據(jù)集目錄??

pathDir?=?os.listdir(filepath)??


R_channel?=0??

G_channel?=0??

B_channel?=0??

for?idx?in?xrange(len(pathDir)):??

????filename?=?pathDir[idx]??

????img?=?imread(os.path.join(filepath,?filename))??

R_channel?=?R_channel?+?np.sum(img[:,:,0])??

G_channel?=?G_channel?+?np.sum(img[:,:,1])??

B_channel?=?B_channel?+?np.sum(img[:,:,2])??


num?=?len(pathDir)?*384?*?512?#?這里(384,512)是每幅圖片的大小,所有圖片尺寸都一樣??

R_mean?=?R_channel?/?num??

G_mean?=?G_channel?/?num??

B_mean?=?B_channel?/?num??


print("R_mean?is?%f,?G_mean?is?%f,?B_mean?is?%f"?%(R_mean,?G_mean,?B_mean))??

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容