Skip to content

author31/scm_nbdev

Repository files navigation

Weakly Supervised Object Localization via Transformer with Implicit Spatial Calibration

Description

  • NBDev version of Weakly Supervised Object Localization via Transformer with Implicit Spatial Calibration with added features
  • LR Scheduler adpated from YOLOX

Install

pip install scm_nbdev

How to train

CUB

python tools_cam/train_cam.py –config_file ./configs/CUB/deit_scm_small_patch16_224.yaml –lr 5e-5

ILSVRC

python tools_cam/train_cam.py –config_file ./configs/ILSVRC/deit_scm_small_patch16_224.yaml –lr 1e-6

How to inference

python tools_cam/test_cam.py –config_file {config_file} –lr {lr}