EfficientNet

2021. 5. 22. 20:13민공지능/딥러닝 & 머신러닝

https://www.tensorflow.org/api_docs/python/tf/keras/applications/EfficientNetB0
https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet

EfiicientNet은 모델 사이즈 별로 B0~B7버전이 있으며 B0이 제일 가벼운 모델이다.

B4부터는 파라미터가 커져도 크게 성능 향상이 없어서 B7이 안될 때는 B4모델을 선택하는 것이 좋을 것 같다.

 

https://arxiv.org/pdf/1905.11946.pdf

EfficientNet에서 activation은 swish를 사용하는데 swish는 깊은 신경망에서 relu보다 좋은 성능을 나타낸다.

 

EfficeintNet B0의 기본 구조

https://lynnshin.tistory.com/13

'민공지능 > 딥러닝 & 머신러닝' 카테고리의 다른 글

Optuna  (0) 2021.05.22
SVM(Support Vector Machine)  (0) 2021.05.22
TTA(Test Time Augmentation)  (0) 2021.05.22
Scikit-Learn의 Scaler  (0) 2021.05.22
Autokeras  (0) 2021.05.06