search

FoundationVision / GLEE

[CVPR2024 Highlight]GLEE: General Object Foundation Model for Images and Videos at Scale
https://glee-vision.github.io/
MIT License
1.02k stars 82 forks source link

修复了没有检测到目标时出现error的问题 #18

Open nightwindnohen opened 5 months ago

nightwindnohen commented 5 months ago 

 in glee
    pred_masks = F.interpolate(mask_pred[None,], size=(padding_size[0], padding_size[1]), mode="bilinear",
  File "/home/user/.local/lib/python3.10/site-packages/torch/nn/functional.py", line 4020, in interpolate
    return torch._C._nn.upsample_bilinear2d(input, output_size, align_corners, scale_factors)
RuntimeError: Non-empty 4D data tensor expected but got a tensor with sizes [1, 0, 270, 480]```