merve HF staff commited on
Commit
15c73ae
โ€ข
1 Parent(s): d4c97d2

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -2
README.md CHANGED
@@ -7,5 +7,13 @@ dataset:
7
  - oxfort-iit pets
8
  license: apache-2.0
9
  ---
10
- ## Image segmentation model on segmenting pets! ๐Ÿถ ๐Ÿ• ๐Ÿฉ
11
- U-Net model on MobileNet weights, trained on Oxford-IIT Pets dataset.
 
 
 
 
 
 
 
 
 
7
  - oxfort-iit pets
8
  license: apache-2.0
9
  ---
10
+ ## Keras semantic segmentation models on the ๐Ÿค—Hub! ๐Ÿถ ๐Ÿ• ๐Ÿฉ
11
+
12
+ Image classification task tells us about a class assigned to an image, and object detection task creates a boundary box on an object in an image. But what if we want to know about the shape of the image? Segmentation models helps us segment images and reveal their shapes. It has many variants. You can host your Keras segmentation models on the Hub.
13
+ Semantic segmentation models classify pixels, meaning, they assign a class (can be cat or dog) to each pixel. The output of a model looks like following.
14
+ ![Raw Output](./raw_output.jpg)
15
+ We need to get the best prediction for every pixel.
16
+ ![Mask](./mask.jpg)
17
+ This is still not readable. We have to convert this into different binary masks for each class and convert to a readable format by converting each mask into base64. You can find the full implementation in pipeline.py.
18
+ ![Binary Mask](./binary_mask.jpg)
19
+