pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

only test people

Open kclch opened this issue 6 years ago • 2 comments

I want to test only people. What should I do? thank you

kclch avatar Apr 09 '19 01:04 kclch

@kclch this YOLOv3 tutorial may help you: https://github.com/ultralytics/yolov3/wiki/Train-Custom-Data

The accompanying repository works on MacOS, Windows and Linux, includes multigpu and multithreading, performs inference on images, videos, webcams, and an iOS app. It also tests to slightly higher mAPs than darknet, including on the latest YOLOv3-SPP.weights (60.7 COCO mAP), and offers the ability to train custom datasets from scratch to darknet performance, all using PyTorch :) https://github.com/ultralytics/yolov3



fourth-archive avatar Apr 11 '19 16:04 fourth-archive

Thank you very much, I mean to use the model trained on the coco dataset, I want to output only the detection of people, I need to change the coco.names file, change the 80 class to 1 class, what else needs to be changed, thank you

kclch avatar Apr 12 '19 07:04 kclch