Skip to content

Commit

Permalink
Minor fixes on model names/paths
Browse files Browse the repository at this point in the history
- added vgg16/vgg19 to readme
- corrected path for googlenet
  • Loading branch information
pjturcot committed May 21, 2016
1 parent 3d77007 commit b09dcd2
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions README.md
Expand Up @@ -24,7 +24,7 @@ All of the necessary code is contained in the file `style.py`. You can try it on
python style.py -s <style_image> -c <content_image> -m <model_name> -g 0
```

The protobufs which come with the vanilla Caffe install aren't quite compatible with this code - working ones have already been added to this repository as a result of this. To get the pretrained models, simply run:
The prototxts which come with the vanilla Caffe install aren't quite compatible with this code - working ones have already been added to this repository as a result of this. To get the pretrained models, simply run:

```
bash scripts/download_models.sh
Expand All @@ -36,7 +36,7 @@ This will grab the convnet models from the links provided in the [Caffe Model Zo
bash scripts/download_models.sh <model_name>
```

Here, `<model_name>` must be one of `vgg`, `googlenet`, or `caffenet`.
Here, `<model_name>` must be one of `vgg16`, `vgg19`, `googlenet`, or `caffenet`.

## Sample

Expand Down
2 changes: 1 addition & 1 deletion style.py
Expand Up @@ -245,7 +245,7 @@ def __init__(self, model_name, use_pbar=True):
# googlenet
elif model_name == "googlenet":
model_file = os.path.join(base_path, "deploy.prototxt")
pretrained_file = os.path.join(base_path, "googlenet_style.caffemodel")
pretrained_file = os.path.join(base_path, "bvlc_googlenet.caffemodel")
mean_file = os.path.join(base_path, "ilsvrc_2012_mean.npy")
weights = GOOGLENET_WEIGHTS

Expand Down

0 comments on commit b09dcd2

Please sign in to comment.