Unverified Commit 0e42539f authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #8 from deepghs/dev/visual 

dev(narugo): make visual boxes in order
parents 177d768e 2be58216

imgutils/detect/visual.py

+1 −1
Original line number Diff line number Diff line
@@ -56,7 +56,7 @@ def detection_visualize(image: ImageTyping, detection: List[Tuple[Tuple[float, f 
    labels = sorted(labels or {label for _, label, _ in detection})

    _colors = list(map(str, rnd_colors(len(labels))))

    _color_map = dict(zip(labels, _colors))

    for (xmin, ymin, xmax, ymax), label, score in detection:

    for _, ((xmin, ymin, xmax, ymax), label, score) in sorted(enumerate(detection), key=lambda x: (x[1][2], x[0])):

        box_color = _color_map[label]

        draw.rectangle((xmin, ymin, xmax, ymax), outline=box_color, width=2)