|
|
|
@@ -138,7 +138,7 @@ def test(data, |
|
|
|
predn = pred.clone() |
|
|
|
scale_coords(img[si].shape[1:], predn[:, :4], shapes[si][0], shapes[si][1]) # native-space pred |
|
|
|
|
|
|
|
# Append to text file(文本) |
|
|
|
# Append to text file(文本发) |
|
|
|
if save_txt: |
|
|
|
gn = torch.tensor(shapes[si][0])[[1, 0, 1, 0]] # normalization gain whwh |
|
|
|
for *xyxy, conf, cls in predn.tolist(): |
|
|
|
|