[일반] 혹시 yolo 모델 사용해신분 계신가요....? 질문좀...ㅜㅜ

[살려줘(210.94)]

yolo v8사용해서 위의 사진과 같이같이 라벨링한 이미지들 학습시켜봤더니 아래사진마냥 결과가 이상하게 나오네요....어떤부분에서부터 잘못한건지 도저히 감이 안잡혀서.......ㅜㅜ 제가 어떤부분에서 실수하신건지 짐작가시는분 계씬가요?

[갸우르(bjukson)]

혹시 pre trained 모델을 사용하지 않고 완전 새로 학습하신게 아닌가요

[살려줘(218.49)]

네네 맞습니다ㅜㅜ pre trained모델을 구해다가 weights경로를 지정해줘야 할까요...??

[갸우르(bjukson)]

네 공식문서 보시면 yolov8s 등의 pretrained 모델의 가중치를 업데이트해서 학습시키는 법이 나와있어요 한번 보시면 좋을 것 같네요

[갸우르(bjukson)]

완전 새로운 데이터로 학습시키려면 데이터셋이 진짜 많이 필요해서 결과가 잘 안 나옵니다

[살려줘(218.49)]

넵! 한번해보겠습니다 정말감사합니다!!!!!!!!!

[익명(218.156)]

pretrained weight를 사용하는게 좋아보여요 다만 team_1과 team_2는 선수들의 유니폼색상을 보고 분류하는 거 같은데 유니폼 색상으로 훈련시켜놓은 학습모델은 없을거에요 그래서 우선 pretrained weight에 글쓴이님 데이터 몇개 집어넣어서 잘 분류하는지 보고 안되면 직접 추가하셔야해요 저같은경우 직접훈련시키는데 각 오브젝트별로 5,000~10,000장 정도 넣어서 훈련했었어요 - dc App