graph-rcnn.pytorch icon indicating copy to clipboard operation
graph-rcnn.pytorch copied to clipboard

How to input image to generate scene graph?

Open khizar-altaf opened this issue 9 months ago • 0 comments

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 main() File "main.py", line 89, in main test(cfg, args) File "main.py", line 37, in test model = build_model(cfg, arguments, args.local_rank, args.distributed) File "/home/oem/GNN/graph-rcnn.pytorch/lib/model.py", line 307, in build_model return SceneGraphGeneration(cfg, arguments, local_rank, distributed) File "/home/oem/GNN/graph-rcnn.pytorch/lib/model.py", line 31, in init self.data_loader_train = build_data_loader(cfg, split="train", is_distributed=distributed) File "/home/oem/GNN/graph-rcnn.pytorch/lib/data/build.py", line 61, in build_data_loader sampler = make_data_sampler(dataset, True if split == "train" else False, is_distributed) File "/home/oem/GNN/graph-rcnn.pytorch/lib/data/build.py", line 15, in make_data_sampler sampler = torch.utils.data.sampler.RandomSampler(dataset) File "/home/oem/.local/lib/python3.8/site-packages/torch/utils/data/sampler.py", line 143, in init raise ValueError(f"num_samples should be a positive integer value, but got num_samples={self.num_samples}") ValueError: num_samples should be a positive integer value, but got num_samples=0

i am using mini_vg dataset

khizar-altaf avatar May 12 '24 16:05 khizar-altaf