diff --git a/model.py b/model.py index 3083762..8f769cd 100644 --- a/model.py +++ b/model.py @@ -251,7 +251,7 @@ def AdapWingLoss(pre_hm, gt_hm): img_dilate = cv2.morphologyEx(img_merge, cv2.MORPH_DILATE, kernel) img_dilate[img_dilate < 51] = 1 # 0*W+1 img_dilate[img_dilate >= 51] = 11 # 1*W+1 - img_dilate = np.array(img_dilate, dtype=np.int) + img_dilate = np.array(img_dilate, dtype=int) img_dilate = img_dilate.transpose(2, 0, 1) dilated = torch.from_numpy(img_dilate).float() if first: diff --git a/tracker.py b/tracker.py index 0854775..f0588c2 100644 --- a/tracker.py +++ b/tracker.py @@ -920,7 +920,7 @@ class Tracker(): x1, y1 = clamp_to_im(center - radius, h, w) x2, y2 = clamp_to_im(center + radius + 1, h, w) offset = np.array((x1, y1)) - lms = (lms[:, 0:2] - offset).astype(np.int) + lms = (lms[:, 0:2] - offset).astype(int) frame = frame[y1:y2, x1:x2] return frame, lms, offset