Preprocessing Code

Open shunlinlu opened this issue 3 years ago • 1 comments

Hi, Dear authors of RelTR:

Your work is so solid that we are so impressed. Is it possible to release the code for preprocessing? so we could reproduce the results on other datasets.

Jul 11 '22 11:07 shunlinlu

Hi,

do you mean the pre-processing for the dataset? For different datasets (OpenImages and Visual Genome) we have different scripts. It is somehow complex. I recommend that you follow the COCO format and notice the dataloader output. I think it is the easiest way:)

For VG-150, I can provide the draft processing code (rebuilding the dataset based on the dataloader output of Neural Motifs). I haven't cleaned it (too busy right now), just for reference: https://drive.google.com/file/d/1vuT0OXla0fhx2ekcLoGs5YZp-VsrRLOM/view?usp=sharing

Jul 11 '22 13:07 yrcong