The training results on the self-built dataset are zero.

[baidingyuan]

@Jingkang50 ，Hi, according to the work you have done, I have created a small dataset specifically for railway scenarios. It only includes two classes of objects: pedestrians and railways. It also includes two types of relationships: "on" and "beside." After training for 100 epochs multiple times, I noticed that the results are consistently zero. I obtained this result when using the predict file to infer an image. As you can see, the PSGTR model I trained predicted the entire image as "person."Can you give me some advice?

[dongdhcugb]

+1