walktree / libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm
452 stars 131 forks source link

detects pedestrians #36

Closed zhanghongruiupup closed 4 years ago

zhanghongruiupup commented 4 years ago

hi,thank your code. I trained a model that only detects pedestrians,nums=1(not 80) Does the code need to be changed?

walktree commented 4 years ago

See #31