Image Segmentation

In the previous chapter, we learned about detecting objects present in images, along with the classes that correspond to the detected objects. In this chapter, we will go one step further by not only drawing a bounding box around the object but also by identifying the exact pixels that contain an object. In addition to that, by the end of this chapter, we will be able to single out instances/objects that belong to the same class.

In this chapter, we will learn about semantic segmentation and instance segmentation by taking a look at the U-Net and Mask R-CNN architectures. Specifically, we will cover the following topics:

  • Exploring the U-Net architecture
  • Implementing semantic segmentation using U-Net
  • Exploring the Mask ...

Get Modern Computer Vision with PyTorch now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.