모바일넷은 핸드폰에서 돌아갈정도로 가볍고, 정확도도 낮지 않아서 잘 쓰이는 인공지능이다.
아래와 같이 쓸 수 있는데
tf.keras.applications.mobilenet_v2.MobileNetV2(
input_shape=None, alpha=1.0, include_top=True, weights='imagenet',
input_tensor=None, pooling=None, classes=1000,
classifier_activation='softmax', **kwargs
)
include_top=False 로 하면 Base Model 만을 가져올 수 있다.
아래는 텐서플로의 공식 사이트이다.
https://www.tensorflow.org/api_docs/python/tf/keras/applications/mobilenet_v2/MobileNetV2?hl=ko#args
tf.keras.applications.mobilenet_v2.MobileNetV2 | TensorFlow Core v2.7.0
Instantiates the MobileNetV2 architecture.
www.tensorflow.org
'인공지능' 카테고리의 다른 글
| COCO (Common Objects in Context) API (0) | 2021.12.29 |
|---|---|
| TensorFlow Object Detection API (0) | 2021.12.29 |
| [인공지능] 원하는 특정(accuracy or loss) 값에서 자동으로 학습 종료하기 (0) | 2021.12.01 |
| [인공지능] 이미지 Classification confusion matrix로 분석하고 heatmap으로 시각화하기 : 실제값이 레이블 인코딩, 예측값이 softmax값일때 (0) | 2021.12.01 |
| [인공지능] Artificial neural network 에서 tensorflow로 이미지 학습시키기 (0) | 2021.12.01 |