inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

두 가지 질문 드립니다.

299

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에 다시 저장한다고 이해하면 되나요?

->

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

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

감사합니다.

 

완독 챌린지 도서 인정 방법에 대한 문의

0

67

2

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

0

77

1

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

0

72

1

수업자료

0

91

2

paperswithcode 서비스 종료 관련 문의

0

396

2

MNS 질문

0

372

1

model.py의 Activation Function

0

586

1

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

0

769

1

object_exists_cell_i 계산식

0

479

0

데이터셋 변경에 대한 질문

0

292

0

reshape 한 후 7*7*30

0

330

1

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

0

359

1

Train시 bounding box의 개수 설정 관련

0

454

1

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

0

334

1

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

0

350

1

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

0

347

1

질문있어서 글 올립니다.

0

344

1

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

0

536

5

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

0

490

1

loss function 관련 질문입니다

0

289

1

loss function 관련

0

257

1

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

0

283

1

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

0

396

1

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

0

392

3