I set "with_visual_mask = True" , But the error

[wyz-gitt]

There is no "ShapeAwareRoIAlign" class in the code. Can you provide the code of this class? I set the mask_roi_layer=dict(type='ShapeAwareRoIAlign',output_size=7,sampling_ratio=2), "openpsg/models/roi_extractors/visual_spatial.py in line 270 and 274, 270: if roi_layers[i].class.name == 'ShapeAwareRoIAlign': 274: roi_feats_t = roilayers[i](feats[i], rois, masks_) : At this time, "ShapeAwareRoIAlign" will be used to extract features, but the class "ShapeAwareRoIAlign" cannot be found. Can you provide the code for "ShapeAwareRoIAlign"?

[Jingkang50]

Please refer to https://github.com/Jingkang50/OpenPSG/issues/8