
pytorch
文章平均质量分 58
西柚与蓝莓
好了阿鹏
展开
-
COCO8 dataset 每种代表什么
parent# Classesnames:0: person1: bicycle2: car5: bus6: train7: truck8: boat13: bench14: bird15: cat16: dog17: horse18: sheep19: cow21: bear22: zebra27: tie30: skis33: kite39: bottle41: cup42: fork43: knife44: spoon45: bowl。原创 2024-09-03 19:17:40 · 465 阅读 · 0 评论 -
Python加载 TorchScript 格式的 ResNet18 模型分类该模型进行预测并输出预测的类别和置信度
读取图像,将图像从 BGR 颜色空间转换为 RGB 颜色空间(因为很多深度学习模型期望输入为 RGB 格式)。(Channel-Height-Width),以符合 PyTorch 的输入要求。最后将处理后的图像转换为 PyTorch 张量,并指定数据类型为。的尺寸,这通常是 ResNet18 模型期望的输入尺寸。(Height-Width-Channel)转换为。可以去Imgnet官网找对应的网站来查看类别。在批量维度上扩展图像,使其形状变为。函数接受一个图像路径作为参数。将图像的像素值归一化到。原创 2024-09-03 18:27:11 · 1009 阅读 · 0 评论 -
NMS写法
param boxes: 检测框列表,形状为 [N, 4],N 为检测框数量,每个检测框为 [x1, y1, x2, y2]:param scores: 检测框对应的置信度得分列表,形状为 [N]:param threshold: 重叠阈值。:return: 保留的检测框索引列表。原创 2024-08-22 16:28:20 · 185 阅读 · 0 评论 -
YOLOv8跑通POSE分类--姿态检测coco8-pos数据集
简单留个备注,方便自己以后查找。原创 2024-08-19 23:04:52 · 1327 阅读 · 0 评论 -
YOLOV8分类任务更换主干mobilenetv2网络
自定义Movilenet_v2类return x。原创 2024-08-16 13:44:54 · 1268 阅读 · 0 评论 -
yolov8自动下载路径与当前文件不一致
可能是第一次注册的时候使用到了pip install ultralytics的时候他给你自动初始化数据集的存放位置。改完后他会自动保存,然后就Ok了。然后自己改一下下载的路径就行了。原创 2024-08-11 11:49:33 · 475 阅读 · 0 评论 -
yolov5主干网络换残差18
在common.py后面加入如下刚刚定义的网络,就是上面的代码的一部分。如此修改网络的yaml文件,yolov5n_res18.yaml。修改yolo网络并运行。需要导包的可以自己导一下。更换网络后的参数如下。原创 2024-08-08 15:05:30 · 635 阅读 · 0 评论 -
yolov5更换主干网络shufflent
这个是YOLOV5的网络。框出来的是yolov5的主干网络。我们用shufflenet_v2的部分替换。可以直接把shufflenet_v2的网络截取出三部分。下图的右边部分是网络shufflenet的官方网络结构,直接使用即可。1.可以先看看shufflenet_v2的网络结构。定义我们自己需要修改的shufflenet类。照着上面的网络对齐修改。原创 2024-08-08 14:25:53 · 896 阅读 · 0 评论 -
yolov5更换MobileNetV3的主干网络
1. 宽度为0.5的mobilenetv2网络的结构图在主干网络上面可以重新定义成三层,编号从0开始如图是MobileNetV3 的网络结构,要想重新定义的画需要保持每次输出图片的大小不变定义MobileNetV3 的代码如下,我们可以分为3层 test_MobileNetV3.pyif n == 0:if n == 1:if n == 2:if n == 3:feature代表的含义。原创 2024-08-08 11:08:10 · 1325 阅读 · 0 评论 -
xml转txt,划分数据集代码
xml转txt的代码 01_xml_to_txt。原创 2024-08-06 11:51:35 · 471 阅读 · 0 评论 -
yolov3 neck部分搭建,以及完整的网络搭建代码,可直接运行
上次提到了搭建了主干网络,这个加上neck就可以构建完整的yolov3网络红色的字体是通道数。当然,你也可以最定义,最后的输出通道数可以根据你的分类数定夺。下面我们一起来实现一下整个代码。原创 2024-08-03 21:24:34 · 335 阅读 · 0 评论 -
代码实现yolov3主干网络,可以直接运行
net=Net()out=net(x)layers=[]net=Net()out=net(x)print(cif)tt.yaml#conf:conf:原创 2024-08-02 10:41:32 · 1596 阅读 · 0 评论 -
PyTorch SummaryWriter TensorBoard 中进行可视化
通常用于在训练过程中记录各种数据,以便在 TensorBoard 中进行可视化。,然后在浏览器中打开显示的 URL 即可查看可视化结果。在 PyTorch 中,原创 2024-07-10 15:46:25 · 641 阅读 · 0 评论 -
交叉熵损失函数的使用目的(很肤浅的理解)
传进去的是(3,5)维度的数据,其中3可以代表有3个图片(数据),5代表有5中类别(0,1,2,3,4这几类)。[ 0.1087, -0.4276, 0.9313, -1.0140, 2.1229]表示预测的是。·····第一个图是第一类的概率是 -0.4276(负数无所谓,举的例子是随机的嘛)是预测值的概率更加接近真实值,让那些真实值对于的概率的类别更加大。····第一个图是第一类的概率是 0.1087。----第一行的第3个值。----第二行的第4个值。----第3行的第2个值。原创 2024-07-10 11:54:44 · 561 阅读 · 0 评论