Update README.md

python setup.py install

```

TODO: coco preprocessing code

#### Retraining

Tensorflow pretrained image classification models available at [https://github.com/tensorflow/models/tree/master/research/slim](https://github.com/tensorflow/models/tree/master/research/slim)

##### ResNet Model (singlelabel) [PDF](https://arxiv.org/pdf/1512.03385.pdf)

* How to retrain ResNet:  

The training is almost the same as with the Singlelabel model.  

Just pass the download link to the ResNet model as hyperparameter. 

Example with tuned hyperparameters:  

```

python3 retrain.py \  

--tfhub_module https://tfhub.dev/google/imagenet/inception_resnet_v2/feature_vector/1 \  

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