KERAS APPLICATION API

이정현·2022년 11월 11일
0

Deep Learning

목록 보기
1/1

KERAS APPLICATION API 공식 사이트

케라스 어플리케이션은 선행학습된 가중치와 함께 사용할 수 있도록 한 딥러닝 모델입니다. 이 모델로 예측, 특성추출, 파인튜닝을 할 수 있습니다.

✅ keras 에서 사용할 수 있는 모델

✅ 사용방법

  1. 이미지 불러오기
!wget -O plane.jpg https://upload.wikimedia.org/wikipedia/commons/1/12/Plane-in-flight.jpg
  1. 모델 불러온 후 요약 보기
from keras.applications.모델명(소문자) import 모델명(대문자), preprocess_input, decode_predictions

아래 스크린샷처럼 사용하고 싶은 모델을 클릭해 들어가서 모델명 참고하기

ex) Xception
from keras.applications.xception import Xception, preprocess_input, decode_predictions
xception = Xception(include_top=True, weights='imagenet',
               input_tensor=None, input_shape=None,
               pooling=None, classes=1000)
xception.summary()

  1. 이미지 학습시키기
img = tf.keras.preprocessing.image.load_img('beaver.jpg', target_size=(299,299))
plt.imshow(img)
x = tf.keras.preprocessing.image.img_to_array(img)
x = x.reshape(1,x.shape[0],x.shape[1],x.shape[2])
x = preprocess_input(x)
preds = xception.predict(x)
print(decode_predictions(preds))

이 방식으로 다른 모델들도 문서 참고하여 파라미터 수정해주면
아주 간단하게 모델을 적용하고 학습시킬 수 있습니다😎

profile
velogon

0개의 댓글