Commit 3b0b952b authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent 5c7da87e
Loading
Loading
Loading
Loading
+12 −1
Original line number Diff line number Diff line
@@ -35,7 +35,18 @@ The pretrained model is retrained onto the 80 categories from coco.
##### Singlelabel Model (InceptionV3) [PDF](https://arxiv.org/pdf/1512.03385.pdf

* How to retrain Singlelabel:  
TODO
Für das training mit default Hyperparameter muss nur das Verzeichnis mit den Bildern übergeben werden:  
python3 retrain.py \
--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/  
  
Veränderte Hyperparameter können auch als Argumente übergeben werden:  
python retrain.py \
--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ \
--how_many_training_steps 4000 \
--flip_left_right True \
--random_scale 20 \
--random_brightness 20 \
--random_crop 20

* **Results:**  
[Trained model (default)](/models/singlelabel_default/)