mmpretrain 설치
mmpretrain 설치
mmpretrain install
mmpretrain 개요
OpenMMLab에서 개발한 프레임워크로 mmpretrain은 image classification 관련된 toolbox이다.
이외에도 object detection, instance segmentation, 3D, OCR 등 다양한 분야를 지원하고 있다.
이 프레임워크는 pytorch기반으로 만들어졌다.
이것을 사용하는 이유는 보다 최신 논문의 내용들을 빠르고, 다양한 사전학습 모델들을 지원하고있다. 아직 딥러닝 영상처리에 대해 잘 모르는 나는 이 프레임워크를 사용해보고, 보다 다양한 프레임워크를 사용해 보기로 계획했다.
mmpretrain install
설치 방법은 mmpretrain documentation에 들어가면 자세하게 설명되어있다.
- https://mmpretrain.readthedocs.io/en/latest/get_started.html
간략하게 설명하면
1. 미니콘다 설치 & 가상환경 생성
2. pytorch 설치
3. mmpretrain 내려받기 & install
추가로 multi-modaliity 관련해서는 추가 설치가 필요한다. 나는 일단 같이 설치해주었다.
설치가 완료되면 아래의 명령어로 제대로 설치가 되었는지 확인이 가능한다.
python demo/image_demo.py demo/demo.JPEG resnet18_8xb32_in1k --device cpu # source install 기준
제대로 설치가 되었다면, 아래 그림과 같이 resnet 모델을 다운받아 추론한 결과를 보여준다.
참고로 colab, docker 등도 지원한다.
docker 사용해보고 docker로 개발환경을 꾸며봐야겠다.
Reference
docs : https://mmpretrain.readthedocs.io/en/latest/get_started.html
github : https://github.com/open-mmlab/mmpretrain?tab=readme-ov-file