thon3 notranslate"><div class="highlight"><pre><span></span><span class="n">image_shape</span> <span class="o">=</span> <span class="p">(</span><span class="mi">256</span><span class="p">,</span><span class="mi">256</span><span class="p">,</span><span class="mi">3</span><span class="p">)</span>
<span class="n">batch_shape</span> <span class="o">=</span> <span class="mi">2</span>
<span class="n">max_objects</span> <span class="o">=</span> <span class="mi">100</span>
<span class="n">train_dataset_path</span> 