Integer printout (#2450)

* Integer printout

* test.py 'Labels'

* Update train.py
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-12 13:35:44 -08:00
1 parent 886f1c03d8
commit f01f3223d5
3 files changed
+4 -4

No files matched your search

+2 -2
View File
@@ -93,7 +93,7 @@ def test(data,
confusion_matrix = ConfusionMatrix(nc=nc)
names = {k: v for k, v in enumerate(model.names if hasattr(model, 'names') else model.module.names)}
coco91class = coco80_to_coco91_class()
s = ('%20s' + '%12s' * 6) % ('Class', 'Images', 'Targets', 'P', 'R', 'mAP@.5', 'mAP@.5:.95')
s = ('%20s' + '%12s' * 6) % ('Class', 'Images', 'Labels', 'P', 'R', 'mAP@.5', 'mAP@.5:.95')
p, r, f1, mp, mr, map50, map, t0, t1 = 0., 0., 0., 0., 0., 0., 0., 0., 0.
loss = torch.zeros(3, device=device)
jdict, stats, ap, ap_class, wandb_images = [], [], [], [], []
@@ -223,7 +223,7 @@ def test(data,
nt = torch.zeros(1)
# Print results
pf = '%20s' + '%12.3g' * 6 # print format
pf = '%20s' + '%12i' * 2 + '%12.3g' * 4 # print format
print(pf % ('all', seen, nt.sum(), mp, mr, map50, map))
# Print results per class