The code is taken from https://github.com/adursun/wsddn.pytorch.git and modified.
Implementation of Weakly Supervised Deep Detection Networks using PyTorch on Google Colab.
Bilen, H., & Vedaldi, A. (2016). Weakly supervised deep detection networks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (pp. 2846-2854).
- Adam optimizer (instead of SGD)
- Spatial regulariser isn't added
VGG16based model is closest toEB + Box Sc.case with model L, which reported 30.4 mAP in the paperAlexNetbased model is closest toEB + Box Sc.case with model S, which reported 33.4 mAP in the paper- Results when
VGG16is used as base model
| aero | bike | bird | boat | bottle | bus | car | cat | chair | cow | table | dog | horse | mbike | person | plant | sheep | sofa | train | tv | mean |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 41.4 | 46.3 | 22.7 | 24.5 | 13.6 | 57.7 | 49.9 | 31.1 | 7.5 | 31.1 | 24.3 | 25.9 | 38.7 | 53.5 | 7.2 | 13.9 | 31.1 | 38.6 | 48.3 | 39.0 | 32.3 |
- Results when
AlexNetis used as base model
| aero | bike | bird | boat | bottle | bus | car | cat | chair | cow | table | dog | horse | mbike | person | plant | sheep | sofa | train | tv | mean |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 38.1 | 41.5 | 27.1 | 18.6 | 10.3 | 48.8 | 47.6 | 36.8 | 1.6 | 25.9 | 28.5 | 30.4 | 39.7 | 46.8 | 15.1 | 12.4 | 28.3 | 32.4 | 44.2 | 44.8 | 30.9 |
Google Colab
!git clone https://github.com/Rupesh-Yadav-1/wsddn.git
!bash /content/wsddn/prepare.sh# for VGG based model
!python /content/wsddn/src/train.py --base_net vgg# for VGG based model
# run `wget "https://www.dropbox.com/s/xyi4hgms6y3ldmj/vgg_epoch_20.pt?dl=1" -P /content/wsddn/states/` to use pretrained weights
python /content/wsddn/src/evaluate.py --base_net vgg --state_path states/vgg_epoch_20.pt