AssertionError assert rois.size(0) > 0

KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”

MIT License

1.06k stars 229 forks source link

AssertionError assert rois.size(0) > 0 #185

Closed theneotopia closed 1 year ago

theneotopia commented 1 year ago

❓ Questions and Help

Hi Kaihua, thanks for your great work. When I am using this codebase with batch_size=1 and SGCLS setting, the following assertion error would occur occasionally. I have checked the dataset while it seems that all image would have at least 1 bounding box, so I couldn't figure out the problem. Any idea? Thanks.

missyoudaisy commented 6 months ago

How did you solve the problem? I also met this occasionally.