Segment prediction labels normalization fix (#10205)

* normalize_segments

* round remove

* swap axes fix
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-11-18 14:19:46 +01:00
1 parent ff6e6e328e
commit 467a57f01b
2 files changed
+11 -8

No files matched your search

+1 -1
View File
@@ -156,7 +156,7 @@ def run(
# Segments
if save_txt:
segments = reversed(masks2segments(masks))
segments = [scale_segments(im.shape[2:], x, im0.shape).round() for x in segments]
segments = [scale_segments(im.shape[2:], x, im0.shape, normalize=True) for x in segments]
# Print results
for c in det[:, 5].unique():