updates
This commit is contained in:
@@ -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__':
|
||||||
|
|||||||
Reference in New Issue
Block a user