search

jwyang / graph-rcnn.pytorch

[ECCV 2018] Official code for "Graph R-CNN for Scene Graph Generation"
727 stars 158 forks source link

ValueError: not enough values to unpack (expected 2, got 1) #110

Closed kwhuang88228 closed 3 years ago

kwhuang88228 commented 3 years ago

I'm getting a "ValueError: not enough values to unpack (expected 2, got 1)" when I run python main.py --config-file configs/sgg_res101_joint.yaml --inference --resume 99999 --algorithm graph-rcnn --visualize. Appreciate any suggestions!

I'm using Python 3.6.12, pytorch 1.0.0, CUDA 10.2.

The exact error message 

Traceback (most recent call last):
  File "main.py", line 92, in <module>
    main()
  File "main.py", line 89, in main
    test(cfg, args)
  File "main.py", line 38, in test
    model.test(visualize=args.visualize)
  File "/data/vision/torralba/scratch/kwhuang/www/graph-rcnn.pytorch/lib/model.py", line 234, in test
    output, output_pred = output
ValueError: not enough values to unpack (expected 2, got 1)

Resolved! I needed to add paths to my object detection and scene graph generation model in configs/sgg_res101_joint.yaml file