Update README.md

athon2 · Apr 4, 2018 · 7407bfe · 7407bfe
1 parent 194a667
commit 7407bfe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -27,4 +27,8 @@ Results are still good
     <img src="imgs/my_seg_results3_OS16.png" width=600></br>
 </p>
 
+How to use this model with custom input shape and custom number of classes:  
+`from model import Deeplabv3`  
+`deeplab_model = Deeplabv3((512,512,3), num_classes=4, last_activation=True, OS=8)`  
+