IT 지식 창고

(Keras) set_image_dim_ordering() -> set_image_data_format

casim 2020. 7. 9. 18:20

tensorflow 2.x가 나오면서, keras 또한 많은 부분이 달라졌는데 하나씩 차이를 찾아보려고 합니다.

 

변경 전/후

setting :

K.set_image_dim_ordering('tf') --> K.set_image_data_format('channels_last')
K.set_image_dim_ordering('th') --> K.set_image_data_format('channels_first')

 

checking :
K.image_dim_ordering() == 'tf' --> K.image_data_format() == 'channels_last'
K.image_dim_ordering() == 'th' --> K.image_data_format() == 'channels_first'