Commit 2b2433fa authored by holzinger's avatar holzinger
Browse files

Update README.md

parent 4f16b84e
Loading
Loading
Loading
Loading
+4 −3
Original line number Diff line number Diff line
@@ -27,8 +27,9 @@ python setup.py install
#### Retraining

Tensorflow pretrained image classification models available at [https://github.com/tensorflow/models/tree/master/research/slim](https://github.com/tensorflow/models/tree/master/research/slim)
The pretrained model is retrained onto the 80 categories from coco.

##### Singlelabel Model [PDF](https://arxiv.org/pdf/1512.03385.pdf)
##### Singlelabel Model (InceptionV3) [PDF](https://arxiv.org/pdf/1512.03385.pdf

* How to retrain Singlelabel:  
TODO
@@ -38,7 +39,7 @@ TODO
[Trained model (+Hyperparameters)](/models/singlelabel_hp/)


##### ResNet Model [PDF](https://arxiv.org/pdf/1512.03385.pdf)
##### ResNet Model (singlelabel) [PDF](https://arxiv.org/pdf/1512.03385.pdf)

* How to retrain ResNet:  

@@ -46,7 +47,7 @@ TODO
[Trained model (default)](/models/resnet_default/)  
[Trained model (+Hyperparameters)](/models/resnet_hp/)

##### Multilabel Model
##### Multilabel Model (InceptionV3)

* How to retrain multilabel: