Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from keras import backend as K
- if K.image_data_format() == 'channels_first':
- input_shape = (3, img_width, img_height)
- else:
- input_shape = (img_width, img_height, 3)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement