• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

mmdetection 학습 중 이미지 파일명 출력 방법에 대해서 문의드립니다.

21.09.16 10:37 작성 조회수 222

0

안녕하세요. mmdetection을 통해서 데이터를 순차적으로 읽어오려고 하고 있습니다. 

관련 설정 변경으로 shuffle 부분을 변경하긴 했는데 

실제로 데이터가 순차적으로 오는지 확인하기 위해서, batch size 크기로 파일명을 

모델 학습 시, 출력을 해보고 싶은데.. 구조가 너무 복잡하다보니 어디에 코드를 넣는게 좋을지 

잘 모르겠습니다. 혹시 어디 부분을 수정하면 좋을지 조언을 받을 수 있을까요?

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

순차적으로 데이터를 읽을 필요는 없습니다만, 꼭 그러셔야 할 일이 있으신지요? 

만약에 그러시더라도 순차적으로 파일명을 가져 올 수 있는 방법은 없을 것 같습니다. mmdetection은 pytorch의 DataLoader를 이용해서 batch size만큼의 image array와 label값을 읽어오는데, 파일명은 반환하지 않습니다. 

shuffle=False를 믿으시는게 좋을 것 같습니다.