span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">ones</span><span class="p">(</span><span class="n">image</span><span class="o">.</span><span class="n">shape</span><span class="p">)</span>
            <span class="n">color_mask</span> <span class="o">=</span> <span class="n">color_list</span><span class="p">[</span><span class="n">mask_color_id</span> <span class="o">%</span> <span class="nb">len</span><span class="p">(</span><span class="n">color_list</span><span cla