baseline 코드 이미지 사이즈에 따른 학습 시간 차이가 없는 현상

합성데이터 기반 객체 탐지 AI 경진대회

채찍피티

2023.05.24 11:46 1,800 조회

안녕하세요, 베이스라인 코드에서 이미지 사이즈를 변경하여 실험을 진행해보는 중입니다. 이미지 사이즈를 변경해서 해봐도 저는 1 epoch 당 다 20분대로 나오는데, 혹시 이유가 뭔지 아시는 분 계실까요...?

댓글 3개

로그인이 필요합니다

comment

0 / 1000

츄릅콩

2023.05.24 13:22

torchvision Faster-RCNN 첫 레이어인 transform에서 resize 작업이 포함되어있어서 그럴 것 같습니다

FasterRCNN(
  (transform): GeneralizedRCNNTransform(
      Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
      Resize(min_size=(800,), max_size=1333, mode='bilinear')
  )
(backbone): BackboneWithFPN(
    (body): IntermediateLayerGetter(
      (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
      (bn1): FrozenBatchNorm2d(64, eps=1e-05)
~~~ 이하 생략