search

walktree / libtorch-yolov3

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

Specified type of detection #31

Closed zhanghongruiupup closed 4 years ago

zhanghongruiupup commented 5 years ago

thanks your work, it's great. I just want to detect a specific kind of object and what to do.(For example, only people are tested.) hope your reply,thanks

walktree commented 5 years ago

if (result_data[i][0] == 0) { // person }

zhanghongruiupup commented 5 years ago

if (result_data[i][0] == 0) { // person }

I will try,Thank you again.