aje6 commited on
Commit
cc72fc9
·
verified ·
1 Parent(s): 7356e47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -244,12 +244,12 @@ def predict(image):
244
  input_name = model.get_inputs()[0].name
245
  input_shape = model.get_inputs()[0].shape
246
 
247
- # print("after model.get_inputs()[0].shape:", image.dims)
248
 
249
  # Resize the image to the model's input shape
250
  image = cv2.resize(image, (input_shape[2], input_shape[3]))
251
 
252
- # print("after cv2.resize:", input_shape)
253
 
254
  # Convert the image to a numpy array and add a batch dimension
255
  if len(input_shape) == 4 and input_shape[0] == 1:
@@ -257,7 +257,7 @@ def predict(image):
257
 
258
  image = image.astype(np.float32)
259
 
260
- # print("np.expand_dims:", input_shape)
261
 
262
  # Perform inference
263
  output = model.run(None, {input_name: image})
 
244
  input_name = model.get_inputs()[0].name
245
  input_shape = model.get_inputs()[0].shape
246
 
247
+ print("start", image.shape)
248
 
249
  # Resize the image to the model's input shape
250
  image = cv2.resize(image, (input_shape[2], input_shape[3]))
251
 
252
+ print("after resize", image.shape)
253
 
254
  # Convert the image to a numpy array and add a batch dimension
255
  if len(input_shape) == 4 and input_shape[0] == 1:
 
257
 
258
  image = image.astype(np.float32)
259
 
260
+ print("after expands/astype", image.shape)
261
 
262
  # Perform inference
263
  output = model.run(None, {input_name: image})