Segment prediction labels normalization fix (#10205)
* normalize_segments * round remove * swap axes fix
This commit is contained in:
1 parent
ff6e6e328e
commit
467a57f01b
2 files changed
+11
-8
No files matched your search
+1
-1
@@ -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():
|
||||
|
||||
Reference in new issue
Block a user