Fix np, uda dev issue (WongKinYiu#1272)

Limit numpy version to avoid issue with tensorboard, conv np.int to just int, fix mismatched device issuecuda dev issue
EdwinKestler · Dec 28, 2022 · 557e383 · 557e383
1 parent 8e9f0b7
commit 557e383
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,7 +2,7 @@
 
 # Base ----------------------------------------
 matplotlib>=3.2.2
-numpy>=1.18.5
+numpy>=1.18.5,<1.24.0
 opencv-python>=4.1.1
 Pillow>=7.1.2
 PyYAML>=5.3.1

diff --git a/utils/datasets.py b/utils/datasets.py
@@ -415,7 +415,7 @@ def __init__(self, path, img_size=640, batch_size=16, augment=False, hyp=None, r
                 x[:, 0] = 0
 
         n = len(shapes)  # number of images
-        bi = np.floor(np.arange(n) / batch_size).astype(np.int)  # batch index
+        bi = np.floor(np.arange(n) / batch_size).astype(int)  # batch index
         nb = bi[-1] + 1  # number of batches
         self.batch = bi  # batch index of image
         self.n = n
@@ -443,7 +443,7 @@ def __init__(self, path, img_size=640, batch_size=16, augment=False, hyp=None, r
                 elif mini > 1:
                     shapes[i] = [1, 1 / mini]
 
-            self.batch_shapes = np.ceil(np.array(shapes) * img_size / stride + pad).astype(np.int) * stride
+            self.batch_shapes = np.ceil(np.array(shapes) * img_size / stride + pad).astype(int) * stride
 
         # Cache images into memory for faster training (WARNING: large datasets may exceed system RAM)
         self.imgs = [None] * n

diff --git a/utils/loss.py b/utils/loss.py
@@ -739,7 +739,7 @@ def build_targets(self, p, targets, imgs):
                 + 3.0 * pair_wise_iou_loss
             )
 
-            matching_matrix = torch.zeros_like(cost)
+            matching_matrix = torch.zeros_like(cost, device="cpu")
 
             for gt_idx in range(num_gt):
                 _, pos_idx = torch.topk(