Commit bd4d7248 authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent a59cb2a5

README.md

+3 −5
Original line number Diff line number Diff line
@@ -36,22 +36,20 @@ The pretrained model is retrained onto the 80 categories from coco. 


* How to retrain Singlelabel:  

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 \  

```  

python3 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/)  

[Trained model (+Hyperparameters)](/models/singlelabel_hp/)