인프런 커뮤니티 질문&답변
작성한 질문수
안녕하세요 교수님
작성
·
397
0
교수님의 강의를 듣고 pytorch로 넘어와서 작업을 하다 tensorflow로 다시 작업을 해야할 상황이 생겼는데,,,,
pytorch 같은 경우 timm이라는 라이브러리를 통해 vit, convnext등 다양한 모델을 이용할 수 있는데 tensorflow같은 경우 어떻게 해야 해당 모델들을 쓸 수 있는지 궁금해서 이렇게 질문을 드립니다 :)
답변 1
0
권 철민
지식공유자
안녕하십니까,
음, timm... tensorflow하는 입장에서는 부럽기도 합니다. 안타깝게도 timm같이 서드파티에서 더 나은 성능을 가진 pretrained 모델을 찾기는 어렵습니다. tensorflow의 내장된 pretrained 모델을 사용할 수 밖에 없습니다.
다만 Transformer 기반의 VIT 같은 경우는 Hugging Face에서 생성된 모듈을 사용하시면 될 것 같습니다.
https://huggingface.co/docs/transformers/model_doc/vit 중간쯤에 TF 기반 VIT 사용법이 있습니다.
감사합니다.
이것저것 검색하는 도중 tfimm(https://github.com/martinsbruveris/tensorflow-image-models) 라이브러리 발견했는데,,,꽤 유용한거같습니다!