Road segmentation with PyTorch

Samuel_Bachorik: <p>My work with Pytorch -<br>
Trained on custom dataset and model.<br>
Github link - project is still not finished !</p><aside class="onebox allowlistedgeneric" data-onebox-src="https://github.com/Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training">
  <header class="source">
      <img src="https://github.githubassets.com/favicons/favicon.svg" class="site-icon" width="32" height="32">

      <a href="https://github.com/Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training" target="_blank" rel="noopener nofollow ugc">GitHub</a>
  </header>

  <article class="onebox-body">
    <div class="aspect-image" style="--aspect-ratio:690/345;"><img src="https://opengraph.githubassets.com/b2504dcd9a04b95a0ef524cd5e23b7b03ebd2fdbd2bce99a1586c0cbafed9cdf/Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training" class="thumbnail" width="690" height="345"></div>

<h3><a href="https://github.com/Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training" target="_blank" rel="noopener nofollow ugc">GitHub - Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training: Road...</a></h3>

  <p>Road segmentation with pytorch. Contribute to Samuel-Bachorik/PyTorch-AI-road-detection-concurrent_training development by creating an account on GitHub.</p>


  </article>

  <div class="onebox-metadata">
    
    
  </div>

  <div style="clear: both"></div>
</aside>
<div class="youtube-onebox lazy-video-container" data-video-id="VmhW1dGFPMw" data-video-title="Test, new neural network model on custom dataset, 6xConv2D layers, Relu Activation" data-provider-name="youtube">
  <a href="https://www.youtube.com/watch?v=VmhW1dGFPMw" target="_blank" rel="noopener nofollow ugc">
    <img class="youtube-thumbnail" src="https://img.youtube.com/vi/VmhW1dGFPMw/maxresdefault.jpg" title="Test, new neural network model on custom dataset, 6xConv2D layers, Relu Activation" width="690" height="388">
  </a>
</div>


rahulbhalley: <p>Amazing work!! I have a few questions:</p>
<ol>
<li>Does it work in real-time?</li>
<li>Can this model generalize well to sky images? I am guessing it will but wanna know what are your thoughts on it.</li>
<li>How long did it took to train and which GPU use used?</li>
</ol>

Samuel_Bachorik: <p>Hello thank you !</p>
<p>Yes yesterday I converted this model to Onnx and it works pretty fast Real Time.</p>
<p>Yes this model will work very good on sky images. - You will need a lot of training data, but I don’t think it’s as much as I do, because the sky doesn’t have as many forms as roads. I think 500 images, cloudy/clear sky/mix will be enough.</p>
<p>I am using RTX 3060 12GB and it took at least 12 hours for good results.</p>

Gabriella_m: <p>(post deleted by author)</p>

rahulbhalley: <p>Thanks man! Btw is Tesla P100 fast or RTX 3060? I couldn’t find the TFlops specs to compare them on NVIDIA website.</p>

Samuel_Bachorik: <p>I am not sure, sorry.</p>

rahulbhalley: <p>Okay, it’s fine. Thanks for code anyways!</p>

arya47: <p>tesla p100 has 9.3 TFLOPS on single precision vs 3060 has 12.8 on single precision</p>

rahulbhalley: <p>Great! Thanks for the information <a class="mention" href="/u/arya47">@arya47</a>.</p>