Update feature extractor to be able to use GoogLeNet #3
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
For AlexNet:
in_size = 227
model_file = 'bvlc_alexnet.caffemodel'
feature_name = 'pool5'
For GoogLeNet:
in_size = 224
model_file = 'bvlc_googlenet.caffemodel'
feature_name = 'pool5/7x7_s1' #aka loss3/fc
No need to manually calculate image_feature_dim.
I tried pool5/7x7_s1 of GoogLeNet instead of pool5 of AlexNet, and got a result which looks "slightly" better (high episode reward sum) than that of AlexNet. See attached.

