Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

applied black

Browse Source
This commit is contained in:
Matthias Bisping 2022-02-04 17:33:07 +01:00
parent 381fe2dbf5
commit d694866e1e
1 changed files with 5 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
fb_detr/utils/box_merging.py
View File

@ -4,7 +4,7 @@ from operator import attrgetter, itemgetter
from frozendict import frozendict from frozendict import frozendict
Rectangle = namedtuple('Rectangle', 'xmin ymin xmax ymax') Rectangle = namedtuple("Rectangle", "xmin ymin xmax ymax")
def make_box(x1, y1, x2, y2): def make_box(x1, y1, x2, y2):
@ -35,7 +35,7 @@ def compute_iou(a, b):
return compute_intersection(a, b) / compute_union(a, b) return compute_intersection(a, b) / compute_union(a, b)
LPBox = namedtuple('LPBox', 'label proba box') LPBox = namedtuple("LPBox", "label proba box")
def less_likely(a, b): def less_likely(a, b):
@ -47,8 +47,7 @@ def overlap_too_much(a, b, iou_thresh):
return iou > iou_thresh return iou > iou_thresh
def filter_contained(lpboxes, iou_thresh=.1): def filter_contained(lpboxes, iou_thresh=0.1):
def remove_less_likely(a, b): def remove_less_likely(a, b):
try: try:
ll = less_likely(a, b) ll = less_likely(a, b)
@ -80,11 +79,7 @@ def lpboxes_to_dict(lpboxes):
boxes, classes, probas = map(list, [boxes, classes, probas]) boxes, classes, probas = map(list, [boxes, classes, probas])
return { return {"bboxes": boxes, "classes": classes, "probas": probas}
"bboxes": boxes,
"classes": classes,
"probas": probas
}
def page_predictions_to_lpboxes(predictions): def page_predictions_to_lpboxes(predictions):
@ -100,22 +95,3 @@ def page_predictions_to_lpboxes(predictions):
def predictions_to_lpboxes(predictions_per_page): def predictions_to_lpboxes(predictions_per_page):
return map(page_predictions_to_lpboxes, predictions_per_page) return map(page_predictions_to_lpboxes, predictions_per_page)