[踩坑全记录]ATSS

ModuleNotFoundError: no module named cv2.cv2

原因：装opencv时装成了opencv，而不是opencv-python
解决：卸载重新装 pip install opencv-python

RuntimeError: Not compiled with GPU support (ml_nms at atss_core/csrc/ml_nms.h:23)

原因：造成这个问题的原因是，conda 中安装的cuda 和ubuntu 系统中的版本不一致，然而，编绎时，用的系统中的cuda 进行编绎的，而调用时则用的conda 中的cuda ，所以一直无法调用。
解决：重装使得两个cuda 版本一致，然后删除build文件，rebuild才可以

原因：编译的时候没有指定GPU
解决：CUDA_VISIBLE_DEVICES=0 python setup.py bulid …

ImportError: cannot import name ‘PILLOW_VERSION’ from ‘PIL’

解决方法

TypeError: only integer tensors of a single element can be converted to an index

原因：pytorch版本问题
解决：

            cv2.putText(
                image, s, (int(x), int(y)), cv2.FONT_HERSHEY_SIMPLEX, .5, (255, 255, 255), 1
            )

pycocotools/_mask.c:4:20: fatal error: Python.h: No such file or directory

compilation terminated.
error: command ‘x86_64-linux-gnu-gcc’ failed with exit status 1
解决:
sudo apt-get install python3.6-dev

https://zhuanlan.zhihu.com/p/65330632