updates
This commit is contained in:
parent
943db40f1a
commit
476724be2d
2
test.py
2
test.py
|
@ -43,7 +43,7 @@ def test(
|
||||||
|
|
||||||
# Dataloader
|
# Dataloader
|
||||||
dataset = LoadImagesAndLabels(test_path, img_size=img_size)
|
dataset = LoadImagesAndLabels(test_path, img_size=img_size)
|
||||||
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=4)
|
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=0)
|
||||||
|
|
||||||
mean_mAP, mean_R, mean_P, seen = 0.0, 0.0, 0.0, 0
|
mean_mAP, mean_R, mean_P, seen = 0.0, 0.0, 0.0, 0
|
||||||
print('%11s' * 5 % ('Image', 'Total', 'P', 'R', 'mAP'))
|
print('%11s' * 5 % ('Image', 'Total', 'P', 'R', 'mAP'))
|
||||||
|
|
Loading…
Reference in New Issue