issues
search
pmz-q
/
est_wassup_03
MIT License
1
stars
0
forks
source link
added: pipeline skeleton 1: training - yolo det, yolo cls
#31
Closed
pmz-q
closed
7 months ago
pmz-q
commented
7 months ago
Run.py
run.py
를 통하여 train 할 수 있습니다.
argument 로는
config_path
를 받고 있으며,
est_wassup_03/configs
안에 yaml 파일 형식으로 저장합니다.
yaml 파일을 읽어서 train / validation 진행 후 결과는
est_wassup_03/results
안에 저장됩니다.
프로젝트명으로 폴더가 자동생성되어 저장되며, 같은 프로젝트 이름으로 여러번 돌리게 되는 경우, 해당 프로젝트 폴더 안에서 1, 2, 3... 등 순차적으로 폴더가 생성됩니다.
config 내용을 모두 저장하고 있습니다.
예시로 yolo detection 과 classification 의 config file 을 생성해두었습니다.
아직 정리가 안 되었고 버그도 많아서, 버그 생기면 바로바로 알려주세요. 감사합니다 정말..
파이프라인 1 마감일 연장
coco 포맷의 training 은 아직 진행중에 있습니다. 월요일 혹은 화요일까지 마무리 될 것 같습니다.
coco 포맷 training 에서 ultralytics yolo 처럼 로깅이랑 플롯 저장하려다보니 시간이 지연되고 있습니다.
정말 죄송하고 양해 부탁드립니다.
또한 pipeline 의 초기작업이다보니 개발양이 많습니다.
2D 파이프라인과 3D 파이프라인을 함께 고려하며 작업하다보니 그렇습니다.
그래도 지금 작업 몰아서 해두면 나중에 2D 랑 3D 파이프라인은 공동작업 할 수 있을 것 같습니다.
아직 코드 정리가 덜 되어서, 이해가 안 되는 부분들을 그때그때 설명드리고, 1차 파이프라인 끝나면 전체 설명 드리겠습니다.
구조가 복잡하진 않습니다. 그냥 개발양이 많을 뿐...
2D 부분 먼저 작업 시작하시는 것은 어떨지 제안드립니다.
코드 읽는 법
run.py 에서부터 하나하나 타고 올라가며 읽는 것을 추천드립니다.
resnet 부분은 아직 코드가 난장판이라 오히려 혼란이 생길 수 있으니 그건 우선 무시해주시기 바랍니다.
PR 올리는 이유는 백업 용 입니다. 그리고 개발중인 구조를 확인하시라고 올립니다.
수정이 필요할 부분, 혹은 버그등은 Kanban 에 카드 생성해주시고 선들(pmz-q) assign 해주세요.
지우님
yolo crop 기능에서 feature 로 저장되게끔 코딩해놔도 될까요?
또한 제가 모델 개발쪽으로 빨라도 월요일(화요일) 까지는 합류하지 못하기 때문에 Project managing 쪽을 다시 정리하고 가야할 것 같습니다.
Bug fix
split train / val 하는 부분에서 버그 있어서 수정하였습니다.
Run.py
run.py
를 통하여 train 할 수 있습니다.config_path
를 받고 있으며,est_wassup_03/configs
안에 yaml 파일 형식으로 저장합니다.est_wassup_03/results
안에 저장됩니다.파이프라인 1 마감일 연장
코드 읽는 법
지우님
Bug fix