search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.71k stars 307 forks source link

when running "bboxes=np.vstack(bbox_result)" in test.py of solo, show the error "TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first." #212

Open xiongcy opened 2 years ago

xiongcy commented 2 years ago

Thanks for your error report and we appreciate it a lot.

Checklist

  1. I have searched related issues but cannot get the expected help.
  2. The bug has not been fixed in the latest version.

Describe the bug A clear and concise description of what the bug is.

Reproduction

  1. What command or script did you run? 
    A placeholder for the command.
  2. Did you make any modifications on the code or config? Did you understand what you have modified?
  3. What dataset did you use?

Environment

  1. Please run python tools/collect_env.py to collect necessary environment infomation and paste it here.
  2. You may add addition that may be helpful for locating the problem, such as
    • How you installed PyTorch [e.g., pip, conda, source]
    • Other environment variables that may be related (such as $PATH, $LD_LIBRARY_PATH, $PYTHONPATH, etc.)

Error traceback If applicable, paste the error trackback here.

A placeholder for trackback.

Bug fix If you have already identified the reason, you can provide the information here. If you are willing to create a PR to fix it, please also leave a comment here and that would be much appreciated!

xiongcy commented 2 years ago

After running "python tools/collect_env.py", we obtain the results as:

TorchVision: 0.9.0 OpenCV: 4.5.5 MMCV: 0.2.16 MMDetection: 1.0.0+unknown MMDetection Compiler: GCC 9.3 MMDetection CUDA Compiler: 11.1

The error described before still exists.