search

WXinlong / SOLO

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

from . import nms_cpu, nms_cuda from .soft_nms_cpu import soft_nms_cpu #122

Closed Shanyaodedanshen closed 3 years ago

Shanyaodedanshen commented 3 years ago

from . import nms_cpu, nms_cuda from .soft_nms_cpu import soft_nms_cpu

help

git clone https://github.com/WXinlong/SOLO.git cd SOLO pip install -r requirements/build.txt pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI" pip install -v -e .

I have successfully installed these command，however, there are no nms_cpu, nms_cuda or soft_nms_cpu.

Backend: pytorch1.1.0

Shanyaodedanshen commented 3 years ago

win10

Shanyaodedanshen commented 3 years ago

I didn't make sure the version of cuda in conda virtual envrionment. I setup pytorch based on the version of cuda in local. Now it's working.

yeungkong commented 3 years ago

@Shanyaodedanshen I am facing the same problem. Could you please specify how you fix this?

Shanyaodedanshen commented 3 years ago

@Shanyaodedanshen I am facing the same problem. Could you please specify how you fix this?

make sure ur cuda version matches the pytorch version.