/Weakly_detector

Tensorflow implementation of "Learning Deep Features for Discriminative Localization"

Primary LanguageJupyter Notebook

Weakly_detector

Tensorflow implementation of "Learning Deep Features for Discriminative Localization"

B. Zhou, A. Khosla, A. Lapedriza, A. Oliva, and A. Torralba Learning Deep Features for Discriminative Localization. Computer Vision and Pattern Recognition (CVPR), 2016. [PDF][Project Page]

Results of Caltech256 Dataset

alt tag

Results of Action40 Dataset

alt tag Object localization using only image-level annotation, without bounding box annotation.

  • If you want to train the model using custom dataset, you need the pretrained VGG Network weights [VGG], which is used in [code].