ZQJiang
ZQJiang
使用的是pytorch版本的centernet代码,主干特征提取网络是resnet50,主干网没动。 在代码: self.backbone = resnet50(pretrain=pretrain) self.decoder = resnet50_Decoder(2048) self.head = resnet50_Head(channel=64, num_classes=num_classes) 中,self.backbone = resnet50(pretrain=pretrain)没动,只更改了self.decoder ,想要迁移使用model_data/centernet_resnet50_voc.pth中的resnet50权重,下面的权重自己训练。 更改了pretrain = True。 却出现错误如下: pretrained_dict = {k: v for k, v in pretrained_dict.items() if...
不知道有没有时间出一期讲解在COCO数据集上训练测试的视频。 比如在2014的训练集上训练,在2014的验证集上测试。 仅仅是一个想法,提议。谢谢。
I installed tf-gpu 1.15.0, my GPU utilization is about 20%, an iter will run for 10s, and a data set will run for about 13 days
Hi, thanks for your great work. I have some questions: **My config:** in centernet.res50.coco.512size: IMS_PER_BATCH=4,NUM_WORKERS=2,.Other config is not change. **My results: All Ap,AR is 0.** thanks
从0开始训练,不使用预训练权重的修改代码: #------------------------------------------------------# #权值文件请看README,百度网盘下载 #------------------------------------------------------# #model_path = 'model_data/voc_weights_resnet.pth' #print('Loading weights into state dict...') #device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') #model_dict = model.state_dict() #pretrained_dict = torch.load(model_path, map_location=device) #pretrained_dict = {k: v...
不知道能不能有一个关于yolo3-keras的ubuntu环境的配置。 比如类似requirement.txt文件