inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

YOLO 구현으로 배우는 딥러닝 논문 구현 with TensorFlow 2.0

두 가지 질문 드립니다.

288

hibernation

작성한 질문수 12

0

TensorFlow 2.0을 이용한 YOLO 논문 구현 chapter의

'train.py - train_step 정의 & for-loop을 이용한 gradient descent 수행 & 파라미터 및 텐서보드(TensorBoard) 로그 저장'

에서 

1분 14초 line 90에

expand_dims을 사용해서 0차원에 dummy dimension 추가하는 code가 있는데 

여기서 왜 dummy dimension을 만드는 건가요?

같은 강의의 

line 279의 if문은

이전에 training 중 의도치 않게 프로그램이 종료되었거나 이전의 학습에 추가적으로 더 학습시키고자 할 때,

저장된 여러 checkpoint 중 마지막 checkpoint에 저장된 model값을  tf.train.Checkpoint의 instance에 다시 저장한다고 이해하면 되나요?

  

딥러닝 tensorflow

답변 1

0

AISchool

안녕하세요~. 반갑습니다.

1.

expand_dims을 사용해서 0차원에 dummy dimension 추가하는 code가 있는데 

여기서 왜 dummy dimension을 만드는 건가요?

->

dummy dimension을 추가한 image array는 결국 tf.keras.applications.InceptionV3 의 input으로 들어가서 처리되게 되는데요.

https://github.com/solaris33/YOLO-v1-tf2/blob/main/model.py#L8

이때 해당 부분에서 인풋으로 기대하는 dimension이 [batch_size, height, width, color_channel]이라서 앞에 dummy로 batch_size=1을 넣어줘서 차원을 맞춰주어서 오류가 나지 않도록 만드는 것입니다.

2.

line 279의 if문은

이전에 training 중 의도치 않게 프로그램이 종료되었거나 이전의 학습에 추가적으로 더 학습시키고자 할 때,

저장된 여러 checkpoint 중 마지막 checkpoint에 저장된 model값을  tf.train.Checkpoint의 instance에 다시 저장한다고 이해하면 되나요?

->

네 이해하신바가 맞습니다~.

공부하시다가 궁금하신점 있으시면 언제든지 편하게 질문주세요~.

감사합니다.

 

130_Transformer.ipynb 오류 대응 부탁드립니다.

0

42

1

<div id="banner">배너 이미지</div> 관련 질문

0

49

1

수업자료

0

63

2

paperswithcode 서비스 종료 관련 문의

0

346

2

22강 코드 call()메서드 is_training -> model(x,False)로 호출시 밸류에러 참고

0

92

1

MNS 질문

0

354

1

model.py의 Activation Function

0

563

1

코랩에서 train.py 실행 시 오류

0

754

1

object_exists_cell_i 계산식

0

469

0

데이터셋 변경에 대한 질문

0

284

0

reshape 한 후 7*7*30

0

322

1

작성한 코드의 저작권과 깃헙 업로드 가능유무에 대한 질문

0

343

1

Train시 bounding box의 개수 설정 관련

0

443

1

backbone network에 대해서 질문 있습니다.

0

317

1

one_hot과 C값에 대해서 질문 드립니다.

0

334

1

class_loss에 대해서 질문 있습니다.

0

330

1

질문있어서 글 올립니다.

0

327

1

안녕하십니까 강의 잘 보고 있습니다

0

508

5

loss.py 의 loss에 사용된 tf.nn.l2_loss 에 대해서 질문 드립니다.

0

475

1

loss function 관련 질문입니다

0

272

1

loss function 관련

0

243

1

NMS 구현 관련 질문있습니다.

0

270

1

confidence가 가장 큰 값으로 IOU 비교 이후 과정에 대해

0

384

1

confidence <= 0.6 이하의 Bounding Box 제거에서 confidence는 class probability가 곱해진 값인지요?

0

370

3