Commit 7ddcbe35 authored by kiegeland's avatar kiegeland
Browse files

Update README.md

parent 9e69870d
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -37,14 +37,14 @@ 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:  
For training with defualt hyperparameters you just need to pass the directory with all the images:
For training with default hyperparameters you need to pass a directory with all the images.  
The directory has to contain subdirectories which have the name of the label and contain all image which are tagged with the label.   
  
```
python3 retrain.py \  
--image_dir /softpro/ss18/caption/cocoapi/train_all_objects/  
```
  
The directory has to contain subdirectories which have the name of the label and contain all image which are tagged with the label.  
You can also pass different hyperparamters as arguments:
```
python3 retrain.py \