Spaces:
Sleeping
Sleeping
Update generic.py
Browse files- generic.py +1 -1
generic.py
CHANGED
@@ -153,7 +153,7 @@ def evaluate(im, device, image_id=None):
|
|
153 |
|
154 |
# get the feature map shape
|
155 |
h, w = conv_features['0'].tensors.shape[-2:]
|
156 |
-
img_np = np.array(im).astype(
|
157 |
|
158 |
fig, axs = plt.subplots(ncols=len(bboxes_scaled), nrows=2, figsize=(22, 7))
|
159 |
for idx, ax_i, (xmin, ymin, xmax, ymax) in zip(keep.nonzero(), axs.T, bboxes_scaled):
|
|
|
153 |
|
154 |
# get the feature map shape
|
155 |
h, w = conv_features['0'].tensors.shape[-2:]
|
156 |
+
img_np = np.array(im).astype(float)
|
157 |
|
158 |
fig, axs = plt.subplots(ncols=len(bboxes_scaled), nrows=2, figsize=(22, 7))
|
159 |
for idx, ax_i, (xmin, ymin, xmax, ymax) in zip(keep.nonzero(), axs.T, bboxes_scaled):
|