DeNA / PyTorch_YOLOv3

Implementation of YOLOv3 in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

problem about NMS in the evaluation.

BucolicWind opened this issue · comments

I just solve all the problems and begin training,everything runs ok,but when I try to make a test on the validation, I found it take much time to get the result given by the cocoapi evaluator. Then I test the single part of test and I found NMS takes about 3.7s for a single class (total class =10) per image,I wonder if there are somewhere I fogot to set?
tips: the code is the raw code and I just change the data for training, training runs ok.