yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021
MIT License
187 stars 34 forks source link

关于AGdataset #15

Closed PrimoWW closed 2 years ago

PrimoWW commented 2 years ago

我发现AG数据集原论文中说relationship一共有26个 并且数据集中也是如此标注的。 但是在你的AGdateset中(把no person and only one frame等不满足情况的去掉后)我发现只有16个relationship在你的dateset中 也即是说 剩下的10种是没有数据的?

yrcong commented 2 years ago

我不太记得filter数据集后剩下多少relationship categories,但在我们去掉没person的frames和帧数极少的视频(比如只有一两帧),会filter掉一些relationships。 AG version 1.0 的person是fasterrcnn标注的,所以会出现frame里没有person的情况,我们当时联系了原论文的作者,他们的操作就是drop这些frame。