![]() If you want to use cpu, please compare which one is faster(darknet or pytorch) version on your own machine.įor Darknet Choose either detectionTkInterGui.py(still in beta) or detectionOpenCvGui.py(you will be able to see a screen with boxex around predicted models )įor better gpu inference you should use pytorchĭownload yolov3-tiny.weights and yolov3-tiny.cfg(you can also use yolov3-tiny-prn_last.weights and yolov3-tiny-prn.cfg for greater speed but lower accuracy) or yolo5s-1.pt for pytorch detection from the following linkĮdit friendlyTeam.txt file to add the classes that you want to detect (0 ,1 for Terrorist, Terrorist Head and 2,3 for Counter Terrorist and CT head)Ĭhange capture params in. If you want to use a gpu, you should stick to the pytorch version(model is not accurate right now) and change "device" field in. ![]()
0 Comments
Leave a Reply. |
Details
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |