uber-research / UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Other
649 stars 120 forks source link

RuntimeError: expected a Variable argument, but got list #147

Open kulkarnikeerti opened 2 years ago

kulkarnikeerti commented 2 years ago

Hi,

I have been facing this issue from two days, I tried to figure that out but could not come up with a solution. I would really appreciate if someone can guide me where I am going wrong.

The below line calls the RCNN with a list and rois as input.

https://github.com/uber-research/UPSNet/blob/aa8434e5a721ed217849607815304f68dfd7720a/upsnet/models/resnet_upsnet.py#L200

RCNN class in turns calls the roi_pooling and which calls ROIAlignFunction. So the problem is that, the first input from RCNN is the list which is forwarded to all the following function calls. However the ROIAlignFunction expects the first input as variable not as a list (Also extracts its shape in ROIAlignFunction class for further use, which throws an error with list as in input)

Can someone guide me if I have missed out something? Or if this list is converted to a variable somewhere on the way? Thanks in advance.