Custom Object Detection Pytorch

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.
Thanks,
Amrit

1 Like

Hey there Check this out!

Is this code for object detection on custom dataset?

no this is not for object detection, this is for custom image classification…

For object detection, you can use ssd, yolov3 etc. These are links for object detection using pytorch :point_down:

For custom object detection using your own dataset look for these repository :point_down:

1 Like

Hi https://github.com/michhar/pytorch-yolo-v3-custom - the link does not work give 404.