diff --git a/lvis/eval.py b/lvis/eval.py index 141e95d..d005de4 100644 --- a/lvis/eval.py +++ b/lvis/eval.py @@ -437,8 +437,8 @@ def accumulate(self): tps = np.logical_and(dt_m, np.logical_not(dt_ig)) fps = np.logical_and(np.logical_not(dt_m), np.logical_not(dt_ig)) - tp_sum = np.cumsum(tps, axis=1).astype(dtype=np.float) - fp_sum = np.cumsum(fps, axis=1).astype(dtype=np.float) + tp_sum = np.cumsum(tps, axis=1).astype(dtype=float) + fp_sum = np.cumsum(fps, axis=1).astype(dtype=float) dt_pointers[cat_idx][area_idx] = { "dt_ids": dt_ids,