Model Zoo

June 27, 2022 ยท View on GitHub

We provide several models refereed in our paper. Here we list the experiments directories in playground/sgg for each model.

Performance

All models use ResNet-101 backbone.

VG

Model(SGGen)mR@50mR@100R@50R@100headbodytail
RelDN6.07.331.435.934.16.61.1
BGNN8.49.829.833.729.811.22.1
BGNN+DETR5.26.522.826.826.55.51.2
SGTR12.616.424.827.927.620.19.4

Experiment Directories

SGG model

  • End-to-End SGG (SGTR)

    • detr.res101.c5.one_stage_rel_tfmer
  • Tranditional two-stage

    • Experiments Directory
      • detr.res101.c5.detr_two_stage_baseline
      • faster_rcnn.res101X.fpn.600size/two_stage_sgg
      • faster_rcnn.res101.c5.600size/two_stage_sgg
    • Models
      • BGNN
      • RelDN
  • Adopt from End-to-End HOI model

    • HOTR
      • hotr
    • AS-Net[TODO]

Object Detection For SGG

  • Experiments Directory
    • detr.res101.c5.multiscale.150e.bs16/det_pretrain
    • faster_rcnn.res101.c5.600size/det_pretrain
    • faster_rcnn.res101X.fpn.coco.600size/det_pretrain
  • Models
    • BGNN
    • RelDN