updates
This commit is contained in:
parent
c9a55a269b
commit
d5db50df8e
2
test.py
2
test.py
|
@ -171,7 +171,7 @@ def test(
|
||||||
map = cocoEval.stats[1] # update mAP to pycocotools mAP
|
map = cocoEval.stats[1] # update mAP to pycocotools mAP
|
||||||
|
|
||||||
# Return results
|
# Return results
|
||||||
return mp, mr, map, mf1, loss
|
return mp, mr, map, mf1, loss / len(dataloader)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
Loading…
Reference in New Issue