您的位置：首页 > 其它

【YOLO】只检测人

2017-08-03 10:42 351 查看

一.修改源代码

cfg/coco.data

classes=1#修改成1
train=/home/pjreddie/data/coco/trainvalno5k.txt
valid=coco_testdev
#valid=data/coco_val_5k.list
names=data/coco.names
backup=/home/pjreddie/backup/
eval=coco


原因：在coco.c中定义的标签，人为第一个，所以改为1.

examples/detector.c

voidtest_detector(char*datacfg,char*cfgfile,char*weightfile,char*filename,floatthresh)函数中
draw_detections(im,l.w*l.h*l.n,thresh,boxes,probs,names,alphabet,1);//最后一个参数修改成1。line=481
voidrun_detector(intargc,char**argv)//该函数中，倒数第三行.line=542
intclasses=option_find_int(options,"classes",1);//最后一个参数修改成1



二.重新编译和测试
1.重新编译

makeclean
make


2.测试

./darknetdetectcfg/yolo.cfgyolo.weightsdata/person.jpg


三.结果



                                            

                                        

                        
                        内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息 
                    

                    

                    

                        

                         标签： 
                                                

                        

                    


                

            


            

                
相关文章推荐

                

                
            

        

        

            

            
            

                

                    
新的分享

                    

                        
                    

                

            


            

                

                    
章节导航