Custom Object Detection Pytorch

Amrit_Das: <p>Is there any examples where a model has been trained and the trained weights are saved. The weights are then used by a different program, which takes an image input and predicts the output based on the weights. I am starting with pytorch. I have worked with Tf and Keras, any help can give me a kick start and is appreciateed.<br>
Thanks,<br>
Amrit</p>

Amrit_Das: <aside class="onebox whitelistedgeneric">
  <header class="source">
      <img src="https://assets-cdn.github.com/favicon.ico" class="site-icon" width="16" height="16">
      <a href="https://github.com/amrit-das/Custom-Model-Training-PyTorch" target="_blank" rel="nofollow noopener">GitHub</a>
  </header>
  <article class="onebox-body">
    <img src="https://avatars1.githubusercontent.com/u/31342979?s=400&amp;v=4" class="thumbnail onebox-avatar" width="60" height="60">

<h3><a href="https://github.com/amrit-das/Custom-Model-Training-PyTorch" target="_blank" rel="nofollow noopener">amrit-das/Custom-Model-Training-PyTorch</a></h3>

<p>PyTorch Framework based API for Custom Model Training - amrit-das/Custom-Model-Training-PyTorch</p>


  </article>
  <div class="onebox-metadata">
    
    
  </div>
  <div style="clear: both"></div>
</aside>

<p>Hey there Check this out!</p>

vaibhavkumar049: <p>Is this code for object detection on custom dataset?</p>

Dhairya_Pant: <p>no this is not for object detection, this is for custom image classification…</p>

Amrit_Das: <p>For object detection, you can use ssd, yolov3 etc. These are links for object detection using pytorch <img src="https://discuss.pytorch.org/images/emoji/apple/point_down.png?v=9" title=":point_down:" class="emoji" alt=":point_down:"></p>
<aside class="onebox whitelistedgeneric">
  <header class="source">
      <img src="https://github.githubassets.com/favicon.ico" class="site-icon" width="16" height="16">
      <a href="https://github.com/amrit-das/pytorch_ssd" target="_blank" rel="nofollow noopener">GitHub</a>
  </header>
  <article class="onebox-body">
    <img src="https://avatars1.githubusercontent.com/u/31342979?s=400&amp;v=4" class="thumbnail onebox-avatar" width="60" height="60">

<h3><a href="https://github.com/amrit-das/pytorch_ssd" target="_blank" rel="nofollow noopener">amrit-das/pytorch_ssd</a></h3>

<p>Pytorch Implemetation of Single Shot Detector (SSD) - amrit-das/pytorch_ssd</p>


  </article>
  <div class="onebox-metadata">
    
    
  </div>
  <div style="clear: both"></div>
</aside>

<aside class="onebox whitelistedgeneric">
  <header class="source">
      <img src="https://github.githubassets.com/favicon.ico" class="site-icon" width="16" height="16">
      <a href="https://github.com/amrit-das/yolov3_pytorch" target="_blank" rel="nofollow noopener">GitHub</a>
  </header>
  <article class="onebox-body">
    <img src="https://avatars1.githubusercontent.com/u/31342979?s=400&amp;v=4" class="thumbnail onebox-avatar" width="60" height="60">

<h3><a href="https://github.com/amrit-das/yolov3_pytorch" target="_blank" rel="nofollow noopener">amrit-das/yolov3_pytorch</a></h3>

<p>Realtime object detection using yolov3 implemented in PyTorch - amrit-das/yolov3_pytorch</p>


  </article>
  <div class="onebox-metadata">
    
    
  </div>
  <div style="clear: both"></div>
</aside>

<p>For custom object detection using your own dataset look for these repository <img src="https://discuss.pytorch.org/images/emoji/apple/point_down.png?v=9" title=":point_down:" class="emoji" alt=":point_down:"></p>
<aside class="onebox whitelistedgeneric">
  <header class="source">
      <img src="https://github.githubassets.com/favicon.ico" class="site-icon" width="16" height="16">
      <a href="https://github.com/michhar/pytorch-yolo-v3-custom" target="_blank" rel="nofollow noopener">GitHub</a>
  </header>
  <article class="onebox-body">
    <img src="https://avatars2.githubusercontent.com/u/13988934?s=400&amp;v=4" class="thumbnail" width="" height="">

<h3><a href="https://github.com/michhar/pytorch-yolo-v3-custom" target="_blank" rel="nofollow noopener">michhar/pytorch-yolo-v3-custom</a></h3>

<p>A PyTorch implementation of the YOLO v3 object detection algorithm for training on custom data with video demo. - michhar/pytorch-yolo-v3-custom</p>


  </article>
  <div class="onebox-metadata">
    
    
  </div>
  <div style="clear: both"></div>
</aside>


Ajeet_Ojha: <p>Hi <a href="https://github.com/michhar/pytorch-yolo-v3-custom" rel="noopener nofollow ugc">https://github.com/michhar/pytorch-yolo-v3-custom</a>  - the link  does not work give 404.</p>