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