Commit a59cb2a5 authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent b5fef550

README.md

+4 −3
Original line number Diff line number Diff line
@@ -36,12 +36,13 @@ 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 \  

--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ \  

--how_many_training_steps 4000 \
@@ -49,9 +50,9 @@ python retrain.py \ 
--random_scale 20 \  

--random_brightness 20 \  

--random_crop 20

```

    

 **Results:**  



* **Results:**  

[Trained model (default)](/models/singlelabel_default/)  

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