인프런 커뮤니티 질문&답변

hibernation님의 프로필 이미지
hibernation

작성한 질문수

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

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

작성

·

221

0

논문에서는 backbone network로 darknet을 사용했는데, 강의 code에서는 keras의 inceptionV3를 사용되어 질문 올립니다.
혹시 darknet을 사용하려면 어떻게 가져와야 하는지 설명해주실 수 있으신가요?
참고할만한 링크라도 남겨주시면 감사드립니다.

답변 1

0

AISchool님의 프로필 이미지
AISchool
지식공유자

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

darknet 백본을 원논문과 동일한 형태로 진행하고 싶으실 경우에는 원 논문에 있는 아래 그림을 참조해서 conv와 pooling을 조합해서 만드시면 됩니다.

pre-trained 된 darknet weight를 사용하고 싶으실 경우에는 아래 링크 등을 참조해서 darknet weight를 tensorflow weight로 변경하는 과정을 진행하셔야 할 것 같습니다.

하지만 아래 repo 자체도 워낙 옛날꺼고 yolo v1이 나온지 오래되서 최신 tensorflow 2.0과 호환되는 오픈소스가 있을지는 잘 모르겠습니다.

https://github.com/jinyu121/DW2TF

darknet weight to tensorflow weight 등의 키워드로 구글링을 해보시면 좋을 것 같습니다.

감사합니다.

hibernation님의 프로필 이미지
hibernation

작성한 질문수

질문하기