search

stevewongv / SSIS

Instance Shadow Detection with A Single-Stage Detector [SSIS & SSISv2] (CVPR 2021 Oral & TPAMI 2022)
Other
90 stars 14 forks source link

problem with train #2

Closed JR-Guo closed 3 years ago

JR-Guo commented 3 years ago

I encounter a problem when i reproduce your code.

in ./adet/modeling/ssis/condinst.py line 96 in forward loss_mask,loss_asso_mask,asso_offset_losses, loss_maskiou, association_masks_loss = self._forward_mask_heads_train(proposals, mask_feats, gt_instances)

not enough values to upack (expected 5, got 3) Do you have any idea about this problem?

stevewongv commented 3 years ago

Thank you for your report. I have fixed this in the latest commit.