擅长:python、mysql、java
<p>这里有一个简洁的矢量化方法,没有那些错误检查-</p>
<pre><code>def unique_map_pixels_vectorized(imgs):
N,H,W = len(imgs), imgs.shape[2], imgs.shape[3]
img2D = imgs.transpose(0, 2, 3, 1).reshape(-1,3)
ID = np.ravel_multi_index(img2D.T,img2D.max(0)+1)
_, firstidx, tags = np.unique(ID,return_index=True,return_inverse=True)
return tags.reshape(N,H,W), img2D[firstidx]
</code></pre>
<p>运行时测试和验证-</p>
^{pr2}$