Merge branch 'ai/develop' of https://lab.ssafy.com/s11-s-project/S11P21S002 into ai/feat/118-segmentation-autolabel

ai/README.md

@@ -1,15 +1,22 @@
 # FastAPI를 이용한 AI 모델 관련 API
 
-## app
+## conda 환경 세팅
+```bash
+conda env create -f environment.yml
+conda activate worlabel_ai_env
+```
 
-### api
-- fastapi 호출 함수 정의
+## FastAPI Project 구조
 
-### schemas
+### app/api
+- api 호출 라우터 정의
+
+### app/schemas
 - api의 request/response 등 Pydantic 모델 정의
 
-### services
-- 필요한 ai 서비스들을 정의
+### app/services
+- AI 관련 패키지를 이용하는 메서드 정의
 
-### utils
+### app/utils
 - 프로젝트 전역에서 이용하는 formatter 등 정의
+

ai/environment.yml

@@ -0,0 +1,15 @@
+name: worlabel_ai_env
+channels:
+  - conda-forge
+  - nvidia
+  - pytorch
+  - defaults
+dependencies:
+  - python=3.10.10
+  - pytorch
+  - torchvision
+  - torchaudio
+  - pytorch-cuda=12.1
+  - fastapi
+  - uvicorn
+  - ultralytics

ai/requirements.txt

@@ -1,7 +1,12 @@
 fastapi==0.104.1
 uvicorn==0.30.6
-torch==2.4.0 -f https://download.pytorch.org/whl/cpu
-torchaudio==2.4.0 -f https://download.pytorch.org/whl/cpu
-torchvision==0.19.0 -f https://download.pytorch.org/whl/cpu
 ultralytics==8.2.82
 ultralytics-thop==2.0.5
+
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.4.0+cu121
+--extra-index-url https://download.pytorch.org/whl/cu121
+torchaudio==2.4.0+cu121
+--extra-index-url https://download.pytorch.org/whl/cu121
+torchvision==0.19.0+cu121
+