worlabel/ai
2024-10-11 09:11:52 +09:00
..
app Final: ai_final 2024-10-11 09:11:52 +09:00
locust Feat: Locust 세팅 - S11P21S002-165 2024-09-10 17:46:12 +09:00
.gitignore Feat: 백엔드 API 연동 2024-09-23 16:34:07 +09:00
environment.yml Feat: GPU 메모리 상태 응답 2024-10-04 11:36:21 +09:00
README.md Docs: README.md에 resources 폴더 구조 설명 추가 2024-09-17 21:22:56 +09:00
requirements.txt Chore: conda로 의존성 관리 추가 2024-09-04 22:26:22 +09:00

FastAPI를 이용한 AI 모델 관련 API

conda 환경 세팅

conda env create -f environment.yml
conda activate worlabel_ai_env

FastAPI Project 구조

app/api

  • api 호출 라우터 정의

app/schemas

  • api의 request/response 등 Pydantic 모델 정의

app/services

  • AI 관련 패키지를 이용하는 메서드 정의

app/utils

  • 프로젝트 전역에서 이용하는 formatter 등 정의

resources/models

  • yolo 기본 모델 6종(default/pretrained, det/seg/cls) 저장

resources/projects/{project_id}/models

  • 프로젝트별 ai 모델 저장

resources/datasets

  • 훈련 데이터셋 저장