回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我有两张NDArray的照片</p>
<ol>
<li>原始图像</li>
<li>预测面具</李>
</ol>
<p>例如,我想将遮罩不在<code>6</code>的区域“白化”</p>
<p>为了简单起见,我把它们贴成3x3的小图片,每个单元格都是像素的RGB值</p>
<p>原件</p>
<pre><code>[
[1,1,1], [1,5,1], [1,1,1]
[3,3,3], [3,3,3], [3,3,3]
[1,1,1], [5,2,1], [1,1,1]
]
</code></pre>
<p>预测</p>
<pre><code>[
[0, 0, 0]
[6, 6, 6]
[1, 2, 3]
]
</code></pre>
<p>为了做到这一点,我只是在预测中循环,用[0,0,0]替换原来的单元格,以删除我不想要的单元格</p>
<pre><code>for rowIndex, predictedPointRow in enumerate(predict):
for colIndex, predPoint in enumerate(predictedPointRow):
if predPoint is not 6:
img[rowIndex][colIndex] = [0, 0, 0]
</code></pre>
<p>然而,这是痛苦的缓慢。有更好的方法吗</p>
<p>谢谢你</p>