Commit 853a0a64 authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent c15fd14d

README.md

+4 −0
Original line number Diff line number Diff line
@@ -36,11 +36,14 @@ 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:  

  

```

python3 retrain.py \  

--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ \
@@ -50,6 +53,7 @@ python3 retrain.py \ 
--random_brightness 20 \  

--random_crop 20

```

  

* **Results:**  

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

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