updates
This commit is contained in:
parent
943db40f1a
commit
476724be2d
2
test.py
2
test.py
|
@ -43,7 +43,7 @@ def test(
|
|||
|
||||
# Dataloader
|
||||
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
|
||||
print('%11s' * 5 % ('Image', 'Total', 'P', 'R', 'mAP'))
|
||||
|
|
Loading…
Reference in New Issue