Weakly-supervised-object-localization

simple weakly dectector

we rewrite two version:
keras + tensorflow2
pytorch

dataset

cifar10, it can be loaded by keras or pytorch with one line

author's paper

Learning Deep Features for Discriminative Localization

code

using the InceptionV3 to extract conv_features
after 3 iterations,training accuracy is 99.87,testing accuracy is 95.68

Step:
(1):run train.py
(2):run draw_detector.py

also,some tricks are learned from:https://github.com/jazzsaxmafia/Weakly_detector

Results show

Here we just show some samples...

Sample 1:

Sample 2:

Sample 3: