I changed the INPUT_SIZE to 384.
train.py:
flags.DEFINE_string('weights', None, 'pretrained weights')
when i convert_tflite, It reports an error:
conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0])
ValueError: cannot reshape array of size 1645 into shape (64,64,1,1)
I changed the INPUT_SIZE to 384. train.py: flags.DEFINE_string('weights', None, 'pretrained weights')
when i convert_tflite, It reports an error: conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 1645 into shape (64,64,1,1)