Update app.py
Browse files
app.py
CHANGED
|
@@ -280,6 +280,7 @@ def plot_image(image, boxes, image_ori=None):
|
|
| 280 |
def infer(model, img, thresh, iou_thresh, anchors):
|
| 281 |
model.eval()
|
| 282 |
image = np.array(img)
|
|
|
|
| 283 |
image_copy = image.copy()
|
| 284 |
# image = image[np.newaxis, :]
|
| 285 |
augmentations = infer_transforms(image=image)
|
|
|
|
| 280 |
def infer(model, img, thresh, iou_thresh, anchors):
|
| 281 |
model.eval()
|
| 282 |
image = np.array(img)
|
| 283 |
+
print(image.shape)
|
| 284 |
image_copy = image.copy()
|
| 285 |
# image = image[np.newaxis, :]
|
| 286 |
augmentations = infer_transforms(image=image)
|