ZCyueternal / LAH

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Requirements

Please install the following packages:

  • python-3.6.5
  • torch-1.2.0+cu92
  • numpy-1.14.3
  • tqdm-4.26.0
  • torchvision-0.4.0+cu92
  • torchnet-0.0.4

Options

  • lr: learning rate
  • lrp: factor for learning rate of pretrained layers. The learning rate of the pretrained layers is lr * lrp
  • batch-size: number of images per batch
  • image-size: size of the image
  • epochs: number of training epochs
  • HASH_TASK: perform hash task
  • IS_USE_MFB: MFB structure has effect
  • HASH_BIT: How long is the generated hash code

examples

VOC 2007

python demo_voc2007_gcn.py data/voc2007/ --image-size 448 --batch-size 32
--HASH_TASK --IS_USE_MFB --NORMED --IS_USE_IOU -t -v --epochs 15 -HASH_BIT 48

COCO

python demo_coco_gcn.py data/coco/ --image-size 448 --batch-size 32
--HASH_TASK --IS_USE_MFB --NORMED --IS_USE_IOU -t -v --epochs 15 --HASH_BIT 48

MIRFLICKR25k

python demo_mirflickr25k_gcn.py data/mirflickr25k/ --image-size 448 --batch-size 32
--HASH_TASK --IS_USE_MFB --NORMED --IS_USE_IOU -t -v --epochs 15 --HASH_BIT 48

About


Languages

Language:Python 100.0%