![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch 学习
文章平均质量分 68
diudiu~bo
CV算法
展开
-
YOLO V4代码理解(未完待续)
YOLOV4理解整理1. voc2faster-rcnn.py2.voc_annotation.py3.kmeans_for_anchors.py4.train.py用自己的数据集调试了YOLOV4,下面是按照运行顺序来写的对各函数的理解1. voc2faster-rcnn.py这个函数的功能是生成trainval.txt,train.txt,val.txt和test.txt,并将生成的文件存放在数据集的ImageSets/Main文件夹中,代码及其注释如下import osimport rand原创 2021-08-25 17:15:57 · 635 阅读 · 0 评论 -
Mobilenet-yolov4代码调试
Mobilenet-yolov4代码调试train.py调试问题一问题二问题三一级目录train.py调试问题一KeyError: 'make_five_conv1.0.conv.weight’Debug运行一下,发现发生错误的语句如下(train.py的第210行),该问题的错误原因是因为,下载的模型pretrained_dict_net文件中没有make_five_conv1.0.conv.weight这个权重内容。pretrained_dict = {k: v for k, v in pre原创 2021-07-20 10:43:14 · 1183 阅读 · 0 评论 -
Pytorch训练SSD时出现的警告和错误
assert win is not None, 'Must define a window to update'AssertionError: Must define a window to updat原创 2021-05-08 16:43:53 · 2678 阅读 · 0 评论 -
修改github上的SSD代码在本地运行
@TOC修改github上的SSD代码在本地运行准备数据首先根据README.md安装一些库1.安装visdompip install visdompython -m visdom.server2.下载数据集sh data/scripts/COCO2014.shsh data/scripts/VOC2007.sh sh data/scripts/VOC2012.sh3.下载VOCdevkit下载到data文件夹下,链接:链接:https://pan.baidu.com/s/1C-原创 2021-05-07 15:49:23 · 565 阅读 · 3 评论 -
如何理解神经网络中的Epoch, Batch, iteration三个参数
在学习SSD算法时,对于迭代训练的部分不是很理解,主要原因是对epoch,batch,iteration三个参数所代表的的意义不清楚,只看代码的话和SSD网络图并不能搞明白,后来查了一下几个参数的定义,豁然开朗,下面是对这段代码的详细解析: # 开始迭代训练 for iteration in range(args.start_iter, cfg['max_iter']): # 如果启用了visdom,迭代次数不为0(iteration=0时已经对loss了,不需要再重置),且所原创 2021-05-07 11:20:58 · 902 阅读 · 0 评论 -
visdom一直蓝屏的解决方法
最近在学习关于神经网络的东西,跑了一些代码,在训练SSD代码时,使用到了visdom可视化软件,虽然安装成功,但是visdom一直蓝屏,没有显示,而且解决办法也不太容易查到。以前遇到问题没有总结的习惯,也不太会用CSDN写东西,但是想从今天开始使用CSDN记录一下学习中的问题,不求排版有多好,毕竟目前时间有限,只是想做一个记录,如果以后在遇到同样的或者类似的问题可以返回来看一看,不用再次从网上查一堆资料了。我使用的是ubuntu1804+python,以下是安装及使用中遇到的问题:平常习惯使用con原创 2021-04-29 23:16:54 · 1528 阅读 · 1 评论