이미지 분류 - TensorFlow 작동 방식

Image Classification - TensorFlow algorithm은 이미지를 입력으로 가져와 출력 클래스 레이블 중 하나로 분류합니다. MobileNet, ResNet, Inception 및 같은 다양한 딥 러닝 네트워크 EfficientNet 는 이미지 분류에 매우 정확합니다. 또한 1,100만 개 이상의 이미지와 거의 11 ImageNet,000개의 클래스를 보유한 와 같은 대규모 이미지 데이터 세트에 대해 훈련된 딥 러닝 네트워크도 있습니다. 네트워크가 ImageNet 데이터로 훈련된 후 특정 포커스가 있는 데이터 세트에서 네트워크를 미세 조정하여 보다 구체적인 분류 작업을 수행할 수 있습니다. Amazon SageMaker Image Classification TensorFlow 알고리즘은 TensorFlow Hub에서 사용할 수 있는 많은 사전 훈련된 모델에서 전송 학습을 지원합니다.

훈련 데이터의 클래스 레이블 수에 따라 선택한 사전 훈련된 TensorFlow 허브 모델에 분류 계층이 연결됩니다. 분류 계층은 드롭아웃 계층, 고밀도 계층, 임의 가중치로 초기화되는 2-norm regularizer가 있는 완전 연결 계층으로 구성됩니다. 이 모델에는 드롭아웃 계층의 드롭아웃 비율에 대한 하이퍼파라미터와 고밀도 계층의 L2 정규화 인자가 있습니다. 그러면 전체 네트워크(사전 훈련된 모델 포함)를 미세 조정하거나 새 훈련 데이터의 상위 분류 계층만 미세 조정할 수 있습니다. 이 전이 학습 방법을 사용하면 더 작은 데이터 세트로 훈련할 수 있습니다.