graph-rcnn.pytorch
graph-rcnn.pytorch copied to clipboard
How to input image to generate scene graph?
Hello, i have git clone graph.rcc.pytorch library for scene graph generation.kindly can anyone guide me how to input image to generate scene graph from this code ?I don't want to train.I want to use pretrained model sg_imp_step_ckpt.pth which i have downloaded sg_imp_step_ckpt.pth(511.3 MB) and placed it in a folder where main.py is located But when I give command in Ubuntu terminal as follows: ~/GNN/graph-rcnn.pytorch$ python3 main.py --config-file configs/sgg_res101_step.yaml --inference --resume 99999 --algorithm graph-rcnn --visualize then following error appears:
2024-05-15 21:31:52,262 scene_graph_generation INFO: Namespace(algorithm='graph-rcnn', batchsize=0, config_file='configs/sgg_res101_step.yaml', distributed=False, inference=True, instance=-1, local_rank=0, resume=99999, session=0, use_freq_prior=False, visualize=True)
2024-05-15 21:31:52,263 scene_graph_generation INFO: Loaded configuration file configs/sgg_res101_step.yaml
2024-05-15 21:31:52,263 scene_graph_generation INFO: Saving config into: logs/config.yml
Traceback (most recent call last):
File "main.py", line 92, in
i am using mini_vg dataset