common preprocessing can get mobilnets image size
@@ -106,6 +106,9 @@ def ck_preprocess(i):
# input image size: https://github.com/ctuning/ck-tensorflow/issues/67
if has_dep_env('weights', 'CK_ENV_TENSORFLOW_MODEL_IMAGE_WIDTH'):
IMAGE_SIZE = int(dep_env('weights', 'CK_ENV_TENSORFLOW_MODEL_IMAGE_WIDTH'))
+ else:
+ if has_dep_env('weights', 'CK_ENV_MOBILENET_RESOLUTION'):
+ IMAGE_SIZE = int(dep_env('weights', 'CK_ENV_MOBILENET_RESOLUTION'))
else:
return {'return': 1, 'error': 'Only TensorFlow model packages are currently supported.'}
