Skip to content

Conversation

kuwajima
Copy link

  • Feature extractor is initialized by network file and feature (layer) name.
    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
  • Feature dim is automatically calculated with input image sizes.
    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.
alexnet_episode_reward_sum
googlenet_episode_reward_sum

@masayoshi-nakamura
Copy link
Contributor

Thank you! I reviewed your code and did experiments.
It is good to be able to choose CNNs. I'm making that config system and will marge your code. Please wait until finished.

matsuren added a commit to Wanwannodao/yosenabe that referenced this pull request Jan 18, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants