WwYc commited on
Commit
9825a89
1 Parent(s): 0100ec9

Update generic.py

Browse files
Files changed (1) hide show
  1. 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(np.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):
 
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):