- 博客(7)
- 收藏
- 关注
RSS订阅原创 使用python将视频文件分帧为图片文件并保存在指定文件夹
最近在做目标检测训练，其中有一些视频的数据集，需要先将视频分帧为图片，然后进行标注和训练。记得以前写过一个脚本，但是今天找不到了，于是准备重新写一个，记录一下。在网上看了一些代码，发现有点麻烦，只能一个视频一个视频的处理，每次都需要改路径，很麻烦。我比较懒，还是自己写一个比较方便。废话不多说，直接上代码。使用的时候只需要直接实例化这个类，然后调用就行了（注意，保存的路径不要放在视频路径下，不然会报错）
2023-03-23 17:29:46
1301
5
原创 yolov5热力图可视化grad-cam踩坑经验分享
最近在做热力图的可视化，网上搜了很多的资料，但是大部分都是需要在原网络结构上进行修改，非常的不方便。最后在网上找到一位博主分享的即插即用的模块，觉得效果还可以，但是中间有些细节，需要注意。将yolo.py文件中，detectionmodel()函数的forward()函数中，true改为false。backward_type：反向传播的方式，可以是以conf的loss传播，也可以class的loss传播。method：选择grad-cam方法，这里是提供了几种，可能对效果有点不一样，可以都尝试一下。
2023-03-09 10:41:11
5276
43
原创 使用YOLOV5训练自己的数据集
1.准备数据集使用labelimg将数据集中需要识别的部位框出来2.划分数据集，训练集编写代码，自动划分，以及将VOC格式转为YOLO格式import xml.etree.ElementTree as ETimport osimport randomfrom shutil import copyfileimport cv2classes = ["gas", "fire"]TRAIN_RATIO = 0.8def clear_hidden_files(path):
2023-03-04 09:26:21
303
原创 如何判断训练中的模型已经收敛
3.模型收敛：train loss 下降趋于稳定，val loss 下降也趋于稳定，且train loss 的值与val loss 的值相差不大。5.train loss上升，val loss上升：网络结构设计有问题，或者训练参数设置不当等，及时停止学习，调整代码。4.train loss稳定，val loss稳定：可能已经收敛，或者学习遇到瓶颈，可以调小学习率试试。1.train loss 下降，val loss下降： 表明网络还在学习。2. train loss下降，val loss稳定：网络过拟合。
2023-02-25 20:59:07
16239
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人