Unverified Commit 4713332c authored by CSautier's avatar CSautier Committed by GitHub
Browse files

fixed deprecation warning evaluation ONCE (#1210)

parent 028ed72e
import numpy as np
def compute_split_parts(num_samples, num_parts):
part_samples = num_samples // num_parts
remain_samples = num_samples % num_parts
......@@ -10,12 +11,14 @@ def compute_split_parts(num_samples, num_parts):
else:
return [part_samples] * num_parts + [remain_samples]
def overall_filter(boxes):
ignore = np.zeros(boxes.shape[0], dtype=np.bool) # all false
ignore = np.zeros(boxes.shape[0], dtype=bool) # all false
return ignore
def distance_filter(boxes, level):
ignore = np.ones(boxes.shape[0], dtype=np.bool) # all true
ignore = np.ones(boxes.shape[0], dtype=bool) # all true
dist = np.sqrt(np.sum(boxes[:, 0:3] * boxes[:, 0:3], axis=1))
if level == 0: # 0-30m
......@@ -30,12 +33,13 @@ def distance_filter(boxes, level):
ignore[flag] = False
return ignore
def overall_distance_filter(boxes, level):
ignore = np.ones(boxes.shape[0], dtype=np.bool) # all true
ignore = np.ones(boxes.shape[0], dtype=bool) # all true
dist = np.sqrt(np.sum(boxes[:, 0:3] * boxes[:, 0:3], axis=1))
if level == 0:
flag = np.ones(boxes.shape[0], dtype=np.bool)
flag = np.ones(boxes.shape[0], dtype=bool)
elif level == 1: # 0-30m
flag = dist < 30
elif level == 2: # 30-50m
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment