티스토리 뷰

Research (연구 관련)

detectron2 install

홍돌 2020. 6. 23. 20:28

Update. 2021. 2.19.

1. detectron2 github에서 코드 clone, python path에 densepoe 추가

sys.path.insert(0, '/path/to/detectron2/projects/DensePose/')

2. colab 가이드대로 pyyamml, detectron2를 pip install. 이 때 detectron2의 torch버전이 자기 torch버전과 같도록 주의할 것. 난 1.6을 쓰니까 1.6으로 바꿔줌

3. pip install av

끝.


기본적으로 virtualenv 가상환경에서 https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5#scrollTo=9_FzH13EjseR에 나온대로 따라하면 된다.

하지만 ubuntu에 python설정이 잘 되어있지않다면, pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

에서 에러가 날 수 있는데, 잘 보면

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

라고 뜰 것이다. 이것을 구글링해보면 https://stackoverflow.com/questions/21530577/fatal-error-python-h-no-such-file-or-directory가 나오는데, sudo apt-get install python3-dev # for python3.x installs 를 해주면 된다.

PS)

최근에 깃헙이 업데이트되었는데 좀 불안정한 것 같다. module이 안깔려 있다거나, 불필요하면서 에러를 일으키는 import를 한다거나 하는 코드가 있다.

하나하나 다시 깔거나 import를 지워주면 해결된다. 얼마 안 걸린다.

공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/05   »
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
글 보관함