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='
|
42 |
shared_head=dict(
|
43 |
-
type='
|
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='
|
52 |
roi_layer=dict(type='RoIPool', output_size=7),
|
53 |
out_channels=1024,
|
54 |
featmap_strides=[16]),
|
55 |
relation_head=dict(
|
56 |
-
type='
|
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,
|