def test_box_from_xyxy_vertical_line():
"""Test to see assertion thrown with invalid points (vertical line)"""
xy1 = np.array([2369, 2975])
xy2 = np.array([2369 + 70, 2975])
with pytest.raises(AssertionError):
_ = BoundingBox.from_xyxy(xy2, xy1)
with pytest.raises(AssertionError):
_ = BoundingBox.from_xyxy(xy1, xy2)
def test_bounding_to_xyxy():
"""Simple case to see if the box can be converted to xyxy"""
tl = np.array([1.0, 1.0])
tr = np.array([2.0, 1.0])
bl = np.array([1.0, 3.0])
br = np.array([2.0, 3.0])
unordered_points = np.array([tl, br, bl, tr])
box = BoundingBox(unordered_points)
xy1, xy2 = box.to_xyxy()
assert np.array_equal(tl, xy1)
assert np.array_equal(br, xy2)
def test_bounding_to_xywh():
"""Simple case to see if the box can be converted to xywh"""
tl = np.array([1.0, 1.0])
tr = np.array([2.0, 1.0])
bl = np.array([1.0, 3.0])
br = np.array([2.0, 3.0])
unordered_points = np.array([tl, br, bl, tr])
box = BoundingBox(unordered_points)
xy, w, h = box.to_xywh()
assert np.array_equal(tl, xy)
assert w == 1.0
assert h == 2.0
def parse_team_yash_alternate(filename):
yash_results = {}
for line in open(filename).readlines():
idx = int(line.split(": ")[0])
rest_objs = [int(val) for val in line.split(": ")[1].split(" ")]
key = "test-%d.png" % idx
n_bbox = rest_objs.pop(0)
#print idx, n_bbox, rest_objs
for i in xrange(n_bbox):
top_y = rest_objs.pop(0)
top_x = rest_objs.pop(0)
width = rest_objs.pop(0)
height = rest_objs.pop(0)
if len(rest_objs):
confidence = rest_objs.pop(0)
else:
print "Warning: no confidence for", idx
confidence = 0
yash_results[key] = yash_results.get(key, []) + ([
BoundingBox(top=top_y,
left=top_x,
width=width,
height=height,
confidence=confidence)
])
return yash_results
def test_bounding_box_center():
xywh = np.array([2369, 2975, 74, 201])
xy = xywh[:2]
w = xywh[2]
h = xywh[3]
bbox = BoundingBox.from_xywh(xy, w, h)
assert np.array_equal(bbox.center, np.array([2406, 3075]))
def gt_to_bbox_densematrix(gt):
return BoundingBox(
top=gt['y'],
left=gt['x'],
width=gt['width'],
height=gt['height'],
confidence=gt['confidence'] if 'confidence' in gt else None,
)
def test_bounding_to_yolo_small_box():
"""Make sure YOLO conversion works correctly with a small bounding box"""
image_dim = np.array([3648, 5472])
xy1 = np.array([4360, 971])
xy2 = np.array([4397, 998])
bbox = BoundingBox.from_xyxy(xy1, xy2)
yolo = bbox.to_yolo(image_dim)
assert np.allclose(
yolo, np.array([0.8000731, 0.26973684, 0.0067617, 0.00740132]))
image_dim = np.array([3456, 4608])
xy1 = np.array([3155, 3432])
xy2 = np.array([3234, 3455])
bbox = BoundingBox.from_xyxy(xy1, xy2)
yolo = bbox.to_yolo(image_dim)
assert np.allclose(
yolo, np.array([0.69314236, 0.99623843, 0.0171441, 0.00665509]))
def test_box_from_xyxy():
"""Test to see if bounding box can be created from XYXY format"""
xy1 = np.array([2369, 2975])
xy2 = np.array([2369 + 40, 2975 + 70])
bbox = BoundingBox.from_xyxy(xy1, xy2)
xy, w, h = bbox.to_xywh()
assert np.array_equal(xy, xy1)
assert w == 40
assert h == 70
def test_box_from_xywh():
"""Test to see if bounding box can be created from XYWH format"""
xywh = np.array([2369, 2975, 74, 78])
xy = xywh[:2]
w = xywh[2]
h = xywh[3]
bbox = BoundingBox.from_xywh(xy, w, h)
xyxy = bbox.to_xyxy()
assert np.array_equal(xyxy, np.array([[2369, 2975], [2369 + w, 2975 + h]]))
def test_bounding_to_yolo_tl_at_zero():
"""Make sure YOLO conversion works correctly with a small bounding box"""
image_dim = np.array([3648, 5472])
xy1 = np.array([4360, 0])
xy2 = np.array([4397, 998])
bbox = BoundingBox.from_xyxy(xy1, xy2)
yolo = bbox.to_yolo(image_dim)
assert np.allclose(
yolo, np.array([0.8000731, 0.13678728, 0.0067617, 0.27357456]))
def parse_team_yash(filename):
yash_results = {}
for line in open(filename).readlines():
idx = int(line.split(":")[0])
key = "test-%d.png" % idx
for (y_s, x_s) in re.findall(r"""\(([0-9]*) ([0-9]*)\)""", line):
yash_results[key] = yash_results.get(key, []) + ([
BoundingBox(top=int(y_s), left=int(x_s), width=100, height=40)
])
return yash_results
def test_bounding_box_init():
"""Make sure that the points are sorted correctly"""
tl = np.array([1.0, 1.0])
tr = np.array([2.0, 1.0])
bl = np.array([1.0, 3.0])
br = np.array([2.0, 3.0])
unordered_points = np.array([tl, br, bl, tr])
ordered_points = np.array([tl, tr, br, bl])
box = BoundingBox(unordered_points)
assert np.array_equal(box.points, ordered_points)
def test_box_from_xyxy_swapped_order():
"""Test to see if bounding box can be created from XYXY
format when top_left and bottom_right are swapped"""
xy1 = np.array([2369, 2975])
xy2 = np.array([2369 + 40, 2975 + 70])
bbox = BoundingBox.from_xyxy(xy2, xy1)
xy, w, h = bbox.to_xywh()
assert np.array_equal(xy, xy1)
assert w == 40
assert h == 70
def test_bounding_to_yolo_invalid_points():
"""Test assertion is thrown when creating YOLO
coordinates with out of bound point"""
image_dim = np.array([5472, 3648])
xy1 = np.array([3640, 78])
xy2 = np.array([3650, 100])
bbox = BoundingBox.from_xyxy(xy1, xy2)
with pytest.raises(AssertionError):
_ = bbox.to_yolo(image_dim)
def test_bounding_from_yolo():
image_dim = np.array([5472, 3648])
xy = np.array([0.65953947, 0.55080409])
w = 0.02028509
h = 0.01425439
bbox = BoundingBox.from_yolo(xy, w, h, image_dim)
output_xy, output_w, output_h = bbox.to_xywh()
assert np.array_equal(np.array([2369, 2975]), output_xy)
assert output_w == 74
assert output_h == 78
def test_bounding_to_yolo():
"""Test to see if bounding box can be converted to YOLO"""
image_dim = np.array([5472, 3648])
xywh = np.array([2369, 2975, 74, 78])
xy = xywh[:2]
w = xywh[2]
h = xywh[3]
bbox = BoundingBox.from_xywh(xy, w, h)
correct = np.array([0.65953947, 0.55080409, 0.02028509, 0.01425439])
yolo = bbox.to_yolo(image_dim)
assert np.allclose(yolo, correct)
record['hpu']['boxes'])
for record in json
}
################################
## Parse AnishShah's team's data
AnishShah_csv = pandas.io.parsers.read_csv(
"StudentData-UIUC-Cars/Team AnishShah/CPU.csv")
AnishShah_CPU = {}
for i, row in AnishShah_csv.loc[1:].iterrows():
key = row['name'].replace("pgm", "png")
if row['x2'] != '0' and row['x1'] != '0':
AnishShah_CPU[key] = AnishShah_CPU.get(key, []) + ([
BoundingBox(
top=int(row['y1']),
height=int(row['y2']) - int(row['y1']),
left=int(row['x1']),
width=int(row['x2']) - int(row['x1']),
)
])
AnishShah_csv = pandas.io.parsers.read_csv(
"StudentData-UIUC-Cars/Team AnishShah/HPU.csv")
AnishShah_HPU = {}
for i, row in AnishShah_csv.loc[1:].iterrows():
key = row['name'].replace("jpg", "png")
if row['x2'] != '0' and row['x1'] != '0':
AnishShah_HPU[key] = AnishShah_HPU.get(key, []) + ([
BoundingBox(
top=int(row['y1']),
height=int(row['y2']) - int(row['y1']),
left=int(row['x1']),
def test_box_from_xyxy_same_point():
"""Test to see assertion thrown with invalid points (points equal)"""
xy1 = np.array([2369, 2975])
with pytest.raises(AssertionError):
_ = BoundingBox.from_xyxy(xy1, xy1)
