我正在嘗試在圖片中找到特定的顏色(或更可能的顏色范圍),然后在第二張圖片上重新繪制坐標(biāo)以進(jìn)行進(jìn)一步調(diào)查。由于我了解了很多有關(guān)numpy的知識(shí),所以我開始使用它(我在python中尤其是numpy相當(dāng)新)。因此,我寫了一些對(duì)一種顏色都適用的東西,但是它存在很多顏色的(性能問題),我很確定,我可以通過使用numpy提供的更多功能來解決這一問題?;旧希蚁霘⑺纅or循環(huán)。那是我的代碼(我以后必須添加更多范圍的顏色):import PIL, numpyfrom PIL import Imagedef add_color_range(p_a_c, r,r1,g,g1,b,b1): for ir in range(r, r1+1): for ig in range(g, g1+1): for ib in range(b, b1+1): p_a_c.extend([[ir,ig,ib]]) return p_a_cfor i in range(1): im = Image.open('%*s.bmp'% (1, i)) n_test = numpy.asarray(im) ni_test = numpy.zeros([1050,1680,3],dtype=numpy.uint8) ni_test.fill(255) c=[] c=add_color_range(c,5,10,5,10,5,10) for ic in range(len(c)): ind=numpy.where(numpy.all(n_test==c[ic], axis=-1)) for ii in range(len(ind[0])): ni_test[ind[0][ii],ind[1][ii]]=[0,0,0] im_test = Image.fromarray(ni_test, 'RGB') im_test.save('test_%*s.bmp'% (1, i))
添加回答
舉報(bào)
0/150
提交
取消