zhb10086 commited on
Commit
1b9b099
·
verified ·
1 Parent(s): 9968cfd

Update relation_afford_r101_caffe_c4_1x_regrad_vmrd_metagraspnet_vrd_vg_class_agnostic.py

Browse files
relation_afford_r101_caffe_c4_1x_regrad_vmrd_metagraspnet_vrd_vg_class_agnostic.py CHANGED
@@ -38,9 +38,9 @@ model = dict(
38
  loss_bbox=dict(type='mmdet.L1Loss', loss_weight=1.0)),
39
  roi_head=None,
40
  child_head=dict(
41
- type='invigorate.PairedRoIHead',
42
  shared_head=dict(
43
- type='invigorate.PairedResLayer',
44
  depth=50,
45
  stage=3,
46
  stride=1,
@@ -48,12 +48,12 @@ model = dict(
48
  norm_eval=False,
49
  share_weights=False),
50
  paired_roi_extractor=dict(
51
- type='invigorate.VMRNPairedRoIExtractor',
52
  roi_layer=dict(type='RoIPool', output_size=7),
53
  out_channels=1024,
54
  featmap_strides=[16]),
55
  relation_head=dict(
56
- type='invigorate.BBoxPairHead',
57
  with_avg_pool=True,
58
  roi_feat_size=7,
59
  in_channels=2048,
 
38
  loss_bbox=dict(type='mmdet.L1Loss', loss_weight=1.0)),
39
  roi_head=None,
40
  child_head=dict(
41
+ type='avision.PairedRoIHead',
42
  shared_head=dict(
43
+ type='avision.PairedResLayer',
44
  depth=50,
45
  stage=3,
46
  stride=1,
 
48
  norm_eval=False,
49
  share_weights=False),
50
  paired_roi_extractor=dict(
51
+ type='avision.VMRNPairedRoIExtractor',
52
  roi_layer=dict(type='RoIPool', output_size=7),
53
  out_channels=1024,
54
  featmap_strides=[16]),
55
  relation_head=dict(
56
+ type='avision.BBoxPairHead',
57
  with_avg_pool=True,
58
  roi_feat_size=7,
59
  in_channels=2048,