Skip to content

gmayday1997/pytorch_CAM

Repository files navigation

Discriminative Localization with Image-Level Annotation

Pytorch 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]

Contents

  1. Basic installation
  2. Data and Pretrain Model
  3. Demo
  4. Beyond the demo:Training and Testing
  5. Results of Action40 Dataset

Basic installation

Requirements for pytorch (see: pytorch installation instuctions)

Data and Pretrain Model

Demo

After sucessfully completing Basic installation and Data and Pretrain Model, you will be ready to run the demo.

cd $CAM_ROOT
./demo.py

Beyond the demo: Training and Testing

Train:

set train_flag = True in action40_config.py , then run train_action40.py.

Test:

set train_flag = False in action40_config.py , then run test_action40.py

Results

alt tag alt tag alt tag alt tag alt tag alt tag alt tag

About

The reimplementation of Class Activation Map in pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages