[❓질문] 좀 살려주라 4시간동안 해결 못 했다

[익명(223.62)]

yolov5로 작업중인데
총 데이터가 총 N개라고 쳤을 때
레이블 데이터가

[label0, x_min0, y_min0, width0, height0]
                                    . . .
[labelN, x_minN, y_minN, widthN, heightN]


이렇게 분류돼있음

이 레이블 데이터를 label 변수에 저장했다 했을 때,
내 상식선에서 label[0]을 하면
[label0, x_min0, y_min0, width0, height0]가

나와야 되는데,
[label0,
label1,
label2,
    . . .
labelN]이 슬라이싱 됨…
어떻게 해결해야 될까?

[익명(223.38)]

그 코드를 좀 보여줘

[익명(223.38)]

아님 batch first가 아닌거 아니야?

[익명(223.38)]

shape이 (batch size, 5)가 아니라 (5, batch size) 같은데?

[익명(223.62)]

밖이라 코드는 당장 못 보여줌 ㅠㅠ 캐글에서 받은 이미지(jpeg)/레이블(txt) 파일이 있고 yolo 돌리려면 coco로 변환해야 된대서 그걸 하는 과정중임 for문 돌려서 yaml이든 xml이든 변환하려면 x_min, y_min, width, height가 N개 있어야 되는데 질문한 문제처럼 0번 데이터만 저장돼서 1개짜리 변수가 돼버림ㅠ

[익명(223.62)]

하 말로만 설명하니 님들도 알아듣기 힘들거고 나도 내 문제를 알려주기가 힘드니 참 답답하고만… 캡처라도 해놓을걸

[익명(223.38)]

깃헙이라두 올려줭

[익명(183.105)]

데이터를 불러오는 과정에 잘못된게 있을듯요

[Cytrus(minhyoo1)]

너가 4시간봐도 모르는문제를 코드도안보고 어케고쳐주냐!!!!!! - dc App