본문 바로가기

인공지능

EfficientNet B0 ~ B7 input / output shape(size), params

  eff_b0 eff_b1 eff_b2 eff_b3 eff_b4 eff_b5 eff_b6 eff_b7
input 224, 224, 3 240, 240, 3 260, 260, 3 300, 300, 3 380, 380, 3 456, 456, 3 528, 528, 3 600, 600, 3
output 7, 7, 1280 8, 8, 1280 9, 9, 1408 10, 10, 1536 12, 12, 1792 15, 15, 2048 17, 17, 2304 19, 19, 2560
params 4,049,571 6,575,239 7,768,569 10,783,535 17,673,823 2,851,3527 40,960,143 64,097,687

output 과 params는 tensorflow.keras.applications을 이용하여 include_top = False를 기준으로 작성하였습니다.

input size의 경우 논문을 참고하였습니다.