Keras의 EfficientNet, EfficientNetV2 모델의 파라미터 개수

Keras의 EfficientNet, EfficientNetV2 모델의 파라미터 개수를 확인해봤습니다. 

2024. 3. 18  최초작성

확인한 모델의 파라미터 개수입니다.  오른쪽에는 M단위로 표기하고 있는데 1M은 백만(1,000,000)을 의미합니다.  예를 들어 5.33M은 533만입니다. 

 

EfficientNetB0  5330571 parameters. (5.33M)

EfficientNetB1  7856239 parameters. (7.86M)

EfficientNetB2  9177569 parameters. (9.18M)

EfficientNetB3  12320535 parameters. (12.32M)

EfficientNetB4  19466823 parameters. (19.47M)

EfficientNetB5  30562527 parameters. (30.56M)

EfficientNetB6  43265143 parameters. (43.27M)

EfficientNetB7   66658687 parameters. (66.66M)

EfficientNetV2B0  7200312 parameters. (7.20M)

EfficientNetV2B1   8212124 parameters. (8.21M)

EfficientNetV2B2  10178374 parameters. (10.18M)

EfficientNetV2B3  14467622 parameters. (14.47M)

EfficientNetV2S     21612360 parameters. (21.61M)

EfficientNetV2M    54431388 parameters. (54.43M)

EfficientNetV2L     119027848 parameters. (119.03M)

모델의 파라미터를 확인시 다음 코드를 사용했습니다. 

 

from tensorflow.keras.applications import EfficientNetB0 # 모델을 로드합니다. model = EfficientNetB0(weights='imagenet') # 전체 파라미터 개수 계산 total_params = model.count_params() total_params_million = total_params / 1e6 print(f'The model has {total_params} parameters. ({total_params_million:.2f}M)')

관련 링크

 

https://keras.io/api/applications/efficientnet/#efficientnetb0-function 

 

https://keras.io/api/applications/efficientnet_v2/