yolo
-
Docker 환경에서 torch install시 process killed카테고리 없음 2024. 6. 15. 18:14
YOLOv5와 FastAPI를 사용한 프로젝트를 Docker Image Build 도중 에러 발생에러 상황Cannot install nvidia-cufft-cu12.ERROR: failed to solve: process "/bin/sh -c pip install poetry==1.8.3 && poetry install" did not complete successfully: exit code: 137 exit code 137은 OOM(Out Of Memory)가 발생했다는 뜻이며 종합해보면 poetry로 dependency를 install하는 과정에서 OOM이 발생했으며 nvidia관련 패키지를 설치하다 OOM이 발생한 것으로 추정된다. nvidia 관련 패키지를 설치하는 이유는 pytorch hub에..
-
YOLOv4 & Deepsort로 Object Tracking 하기카테고리 없음 2024. 5. 10. 10:56
프로젝트 환경 설정가상환경 설정에는 두 가지 방법이 존재한다. - Virtual Env (venv)python -m venv yolov4-cpu source yolov4-cpu/Scripts/activate pip install -r requirement.txt - Conda# Tensorflow CPU conda env create -f conda-cpu.yml conda activate yolov4-cpu YOLOv4 Tiny Weights 다운로드https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights Object Tracking 과정 인공지능을 크게 학습(Training)과 예측(Infe..