1. 首页
  2. 移动开发
  3. 其他
  4. pytorch ssd:源程序来自于:https 源码

pytorch ssd:源程序来自于:https 源码

上传者: 2021-04-04 18:00:17上传 ZIP文件 20.16MB 热度 9次
本代码在原作者的基础上增加了部分功能,主要用于个人学习ssd流程:补充部分:(1)原程序无法在断网的情况下进行初始化网络,原因在于加载vgg16预训练模型的时候从torchvision.models.vgg16正在加载,需要联网,本程序将model.py中的这三行代码keys())更改为:vgg16_model_path ='pretrained / vgg16-397923af.pth'pretrained_state_dict =火炬负载(vgg16_model_path)pretrained_pa​​ram_names = list(pretrained_state_dict.keys()这三行。并发布百度云地址。 (2)原程序中产生8732个附加框的设置中,fmap_dims,obj_scales,aspect_ratios,这两个变量采用普通字典形式,但是python更新后,普
用户评论