YOLOv5 Custom Training : GPU로 Roboflow 데이터셋을 학습시켜보자 (window11, RTX3060Ti) #1
·
Dev
졸업작품으로 언젠가 해보고 싶었던 AI 모델학습을 직접 해보기로 했음!작년에 선배들 졸업작품 전시회랑 학술제들을 둘러보다가 YOLO라는 이미지 기반 AI모델을 활용해 모델학습을 시켜서사용자 이미지에 매핑하는걸 처음 봤는데 엄청 신기했던 거임잘만 사용하면 웹캠이나 cctv 같은 거에 연동해서 사물인터넷이나 실사용 시스템을 만들 수도 있겠다 싶었음실제로 모델학습에 대한 공부 없이 노 베이스로 박치기해보려고 했는데 유튜브랑 다른 블로그 글 이것저것 봐도 google coleb을 주로 많이 쓰던데, 로컬 pc에서 GPU 써가지고 직접 학습시키는 내용은 많이 없었음계획하고 있는 졸작이 학습한 모델로 백엔드서버랑 연동해서 웹사이트까지 만드는 거라그냥 로컬에 갖고 있으면서 돌리는 게 편할 듯했음 colab.googl..