Scene-Graph-Benchmark.pytorch icon indicating copy to clipboard operation
Scene-Graph-Benchmark.pytorch copied to clipboard

PredCls, SGCls for custom images

Open park-youngjae opened this issue 2 years ago • 1 comments

Thank you for your wonderful work for Scene Graph Generation tasks.

I would like to use your code as a baseline for my research.

In addition, could you provide PredCls and SGCls codes for custom images inference?

Thank you.

park-youngjae avatar Mar 01 '23 03:03 park-youngjae

Check the reademe.md first.

SGDet on Custom Images

Note that evaluation on custum images is only applicable for SGDet model, because PredCls and SGCls model requires additional ground-truth bounding boxes information. To detect scene graphs into a json file on your own images, you need to turn on the switch TEST.CUSTUM_EVAL and give a folder path (or a json file containing a list of image paths) that contains the custom images to TEST.CUSTUM_PATH. Only JPG files are allowed. The output will be saved as custom_prediction.json in the given DETECTED_SGG_DIR.

explorer-shy avatar Apr 25 '23 06:04 explorer-shy