[ 인공지능/프레임워크 or 라이브러리 ]
VGGNet을 사용한 이미지 분류기 실습
2024-03-19 05:33:45
실습 목표 VGGNet을 사용하여 이미지를 학습하고 10개의 카테고리를 갖는 이미지를 분류하는 이미지 분류기를 생성한다. (데이터셋: CIFAR) Pre-training 모델의 사용방법을 이해한다. 문제 정의 VGGNet 구조 살펴보기 주요 코드 1. VGGNet # Model cfg = { 'VGG11': [64, 'M', 128, 'M', 256, 256, 'M', 512, 512, 'M', 512, 512, 'M'], 'VGG13': [64, 64, 'M', 128, 128, 'M', 256, 256, 'M', 512, 512, 'M', 512, 512, 'M'], 'VGG16': [64, 64, 'M', 128, 128, 'M', 256, 256, 256, 'M', 512, 512, 512, 'M..