Ultralytics YOLO는 객체 탐지, 분할, 분류, 자세 추정, 추적 등을 지원하는 컴퓨터 비전용 Python 라이브러리입니다.
huggingface/pytorch-image-models
컴퓨터 비전 작업을 위한 이미지 모델 아키텍처, 사전 학습 가중치, 학습·검증·내보내기 스크립트를 제공하는 PyTorch 라이브러리입니다.
- 스타
- 37,232
- 포크
- 5,217
- 열린 이슈
- 18
- 마지막 커밋
- 11시간 전
어떤 프로젝트인가요?
PyTorch Image Models(timm)는 ResNet, EfficientNet, Vision Transformer(ViT), Swin Transformer, ConvNeXt, MobileNet 계열 등 다양한 이미지 인코더와 백본을 모아 둔 Python 라이브러리이며, 다수의 모델에 사전 학습 가중치가 제공됩니다. 공통 팩토리를 통해 이름으로 모델을 생성하므로, 같은 코드로 ResNet이나 ViT를 불러와 분류에 사용하거나 특징 추출기로 활용할 수 있습니다.
모델 외에도 저장소에는 옵티마이저, 학습률 스케줄러, RandAugment·Mixup·CutMix 같은 데이터 증강 기법과 학습·검증·추론용 예제 스크립트가 포함되어 있습니다. 모델은 분류기를 교체하고, 특징을 추출하며, 다중 스케일 특징 피라미드를 생성하는 일관된 인터페이스를 공유합니다.
이 프로젝트는 PyTorch에서 비전 백본을 재사용하거나 벤치마크하고, ImageNet 학습 결과를 재현하려는 연구자와 엔지니어를 대상으로 합니다. 사전 학습 가중치는 Hugging Face Hub를 통해 배포되며, 라이브러리는 Apache 2.0 라이선스를 따릅니다. 단, ImageNet으로 학습한 가중치에는 해당 데이터셋의 비상업적 이용 조건이 적용됩니다.
핵심 기능
- 01ResNet, EfficientNet, ViT, Swin, ConvNeXt, MobileNet을 비롯한 수백 가지 이미지 아키텍처를 지원하며, 다수의 모델에 사전 학습 가중치가 제공됩니다.
- 02공통 팩토리 함수를 통해 지원되는 모델을 이름으로 생성할 수 있습니다.
- 03AdamW 계열, Lion, Muon 등의 옵티마이저와 학습률 스케줄러가 내장되어 있습니다.
- 04RandAugment, AugMix, Mixup, CutMix 등의 데이터 증강 기법을 지원합니다.
- 05분산 멀티 GPU를 지원하는 학습·검증·추론용 예제 스크립트를 제공합니다.
- 06분류기를 교체하고 다중 스케일 특징 맵을 추출할 수 있는 일관된 API를 제공합니다.
활용 사례
- 사전 학습된 ViT 또는 ConvNeXt를 사용자 지정 이미지 분류 데이터셋에 맞게 파인튜닝합니다.
- 객체 탐지 또는 세그멘테이션 파이프라인에서 백본을 특징 추출기로 사용합니다.
- 동일한 학습 및 평가 설정에서 여러 아키텍처를 벤치마크합니다.
- ImageNet 학습 결과를 재현하거나 배포를 위해 모델을 내보냅니다.
빠른 시작
- 1
PyPI에서 라이브러리 설치하기
pip install timm - 2
사전 학습된 모델을 만들고 README의 빠른 예제로 추론 실행하기
사용 전 알아둘 점
- 실행하려면 PyTorch를 별도로 설치해야 합니다.
- 일부 모델 변형과 가중치는 최신 버전의 PyTorch에 의존합니다.
토픽
자주 묻는 질문
- pytorch-image-models은(는) 무엇인가요?
- 컴퓨터 비전 작업을 위한 이미지 모델 아키텍처, 사전 학습 가중치, 학습·검증·내보내기 스크립트를 제공하는 PyTorch 라이브러리입니다. PyTorch Image Models(timm)는 ResNet, EfficientNet, Vision Transformer(ViT), Swin Transformer, ConvNeXt, MobileNet 계열 등 다양한 이미지 인코더와 백본을 모아 둔 Python 라이브러리이며, 다수의 모델에 사전 학습 가중치가 제공됩니다. 공통 팩토리를 통해 이름으로 모델을 생성하므로, 같은 코드로 ResNet이나 ViT를 불러와 분류에 사용하거나 특징 추출기로 활용할 수 있습니다.
- pytorch-image-models은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- pytorch-image-models은(는) Apache-2.0 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- pytorch-image-models은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. pytorch-image-models의 마지막 커밋은 2026년 10월 20일입니다.
- pytorch-image-models은(는) 어디에 사용할 수 있나요?
- 사전 학습된 ViT 또는 ConvNeXt를 사용자 지정 이미지 분류 데이터셋에 맞게 파인튜닝합니다. 객체 탐지 또는 세그멘테이션 파이프라인에서 백본을 특징 추출기로 사용합니다. 동일한 학습 및 평가 설정에서 여러 아키텍처를 벤치마크합니다. ImageNet 학습 결과를 재현하거나 배포를 위해 모델을 내보냅니다.
- pytorch-image-models는 어떻게 설치하나요?
- PyPI에서 라이브러리 설치하기: pip install timm 사전 학습된 모델을 만들고 README의 빠른 예제로 추론 실행하기
- pytorch-image-models은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/huggingface/pytorch-image-models
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
Ultralytics YOLOv5는 객체 탐지, 인스턴스 분할, 이미지 분류를 지원하며 학습과 내보내기를 제공하는 PyTorch 기반 컴퓨터 비전 툴킷입니다.
컴퓨터 비전 모델의 예측에 영향을 준 이미지 영역을 시각화하는 CAM 기법을 구현한 PyTorch 라이브러리입니다.
최신 텍스트, 비전, 오디오 및 멀티모달 머신러닝 모델을 정의하고 추론과 학습에 활용하는 프레임워크입니다.
대규모 모델의 분산 학습과 추론을 더 쉽고 효율적으로 지원하는 딥러닝 최적화 라이브러리입니다.
코드북 조회 트랜스포머로 사진과 동영상의 흐리거나 품질이 낮고 손상된 얼굴을 복원하는 Python 도구