"n">c</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">3</span><span class="p">):</span>
                <span class="n">img</span><span class="p">[:,</span> <span class="p">:,</span> <span class="n">c</span><span class="p">]</span> <span class="o">=</span> <span class="n">color_mask</span><span class="p">[</span><span class="n">c</span><span class="p">]</span>

            <span class="n">e</span> <span class="o">=</span> <span class="n">decode_segment