基于yolo的视觉检测如何进行二次检测,比如在街道上先识别到汽车,再二次检测汽车是什么牌子的。请求解答,谢谢各位导师。
编写函数,调用两次不同权重的识别函数,对第一次识别汽车的时候进行裁剪
路径改为你放图片的地方:
file_root = '/home/xzh/data/VOCdevkit/VOC2012/allimgs/'
数据集,训练只用voc2012:
train_dataset = yoloDataset(root=file_root,list_file=['voc2012.txt','voc2007.txt'],train=True,transform = [transforms.ToTensor()] )
train_loader = DataLoader(train_dataset,batch_size=batch_size,shuffle=True,num_workers=4)
改为:
train_dataset = yoloDataset(root=file_root,list_file=['voc2012.txt'],train=True,transform = [transforms.ToTensor()] )
train_loader = DataLoader(train_dataset,batch_size=batch_size,shuffle=True,num_workers=4)
这个好像是一次性识别的有这种品牌的模型