Request for guidance

[Duwimpuhwe]

Hi Daniel,

I'm in need of diving in image segmentation and object detection, any quick guidance (course links , video, docs, etc). If the framework is pytorch, that would be super.

Thanks

[mrdbourke]

Hi @Duwimpuhwe ,

For these, I'd suggest either the pre-trained models in PyTorch:

• Segmentation - https://pytorch.org/vision/stable/models.html#semantic-segmentation
• Object detection - https://pytorch.org/vision/stable/models.html#object-detection-instance-segmentation-and-person-keypoint-detection

Or the Ultralytics GitHub (one-stop shop for many PyTorch-based object detection + image segmentation models): https://github.com/ultralytics/ultralytics