fix zero labels (#10820)

update
This commit is contained in:
Laughing authored and GitHub committed 2023-02-04 16:21:30 +04:00
1 parent 064365d868
commit 6a62c94190
1 file changed
+1 -1
+1 -1
View File
@@ -201,7 +201,7 @@ def random_perspective(im,
# Transform label coordinates
n = len(targets)
if n:
use_segments = any(x.any() for x in segments)
use_segments = any(x.any() for x in segments) and len(segments) == n
new = np.zeros((n, 4))
if use_segments: # warp segments
segments = resample_segments(segments) # upsample