pytorch实现yolov3并用coco2017_scratch训练

2024-03-15 18:50

本文主要是介绍pytorch实现yolov3并用coco2017_scratch训练,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

pytorch实现yolov3并用coco2017 scratch训练

用torch.sum而不是mean编写了几个不同的损失计算函数,并使用yolov3-tiny网络进行训练,它是在带有多gpu卡的gpu服务器上进行训练,服务器是网上租来的,所以训练很少的epoch并使用小的batch size来节省资金,所以没有太过调整hyper_params。与之前相比,训练效果好了很多的,这是个人的GitHub

参考资料主要是最后面的几个website:

损失函数主要是这几个, calculate_losses_darknet 训练还是有些不太稳定。所以做了修改的,用iou来算坐标损失而不是用mse来算坐标损失,也就是函数 calculate_losses_darknetRevise, 对应的参考website也就是这个

https://github.com/pjreddie/darknet

这个损失函数 calculate_losses_Alexeydarknet 对应到这个website:

https://github.com/pjreddie/darknet

这个损失函数 calculate_losses_yolofive 对应到这个website:

https://github.com/ultralytics

这个损失函数 calculate_losses_20230730 是我自己写的,使用iou来选择相应的anchor,并且使用了num_scale来平衡不同类数量的差异,使用iou_scale来平衡不同anchor框和label的iou不同

检测的输出展示,这是yolov3-tiny alexeydarknet的输出结果,训练了70个epoch其他的模型还在训练的

参考yolov5s模型写的模型,和训练的codes,训练7个epoch,val2017的mAP是16%,训练了10个epoch,val2017的mAP是21.2%,之后还会训练其他的。

训练使用的数据就是coco2017
分割模型看起来更好些的,bbox的mAP达到了33% model_e46seg_map[0.50716__0.334552]_lnan_2023-09-01.pt

model_e60seg_map[0.485051__0.295817]_lnan_2023-09-03.pt https://www.aliyundrive.com/s/bLnC733gX5C 提取 nu63

pytorch实现yolov3分割segmentation并用coco2017 scratch训练
pytorch实现yolov3分割segment+姿态关节点检测pose keypoint并用coco2017 scratch训练

pytorch实现yolov3分割segmentation并用coco2017 scratch训练

pytorch实现yolov3分割segment+姿态关节点检测pose keypoint并用coco2017 scratch训练

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

https://github.com/ZouJiu1/Pytorch_YOLOV3

参考:

https://github.com/pjreddie/darknet

https://github.com/AlexeyAB/darknet

https://github.com/qqwweee/keras-yolo3

https://github.com/eriklindernoren/PyTorch-YOLOv3

https://github.com/YunYang199/tensorflow-yolov3

https://github.com/mystic123/tensorflow-yolo-v3

https://github.com/Ray-Luo/YOLOV3-PyTorch

https://github.com/DeNA/PyTorch_YOLOv3


https://zhuanlan.zhihu.com/p/648865953

这篇关于pytorch实现yolov3并用coco2017_scratch训练的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/812933

相关文章

Oracle查询优化之高效实现仅查询前10条记录的方法与实践

《Oracle查询优化之高效实现仅查询前10条记录的方法与实践》:本文主要介绍Oracle查询优化之高效实现仅查询前10条记录的相关资料,包括使用ROWNUM、ROW_NUMBER()函数、FET... 目录1. 使用 ROWNUM 查询2. 使用 ROW_NUMBER() 函数3. 使用 FETCH FI

Python脚本实现自动删除C盘临时文件夹

《Python脚本实现自动删除C盘临时文件夹》在日常使用电脑的过程中,临时文件夹往往会积累大量的无用数据,占用宝贵的磁盘空间,下面我们就来看看Python如何通过脚本实现自动删除C盘临时文件夹吧... 目录一、准备工作二、python脚本编写三、脚本解析四、运行脚本五、案例演示六、注意事项七、总结在日常使用

Java实现Excel与HTML互转

《Java实现Excel与HTML互转》Excel是一种电子表格格式,而HTM则是一种用于创建网页的标记语言,虽然两者在用途上存在差异,但有时我们需要将数据从一种格式转换为另一种格式,下面我们就来看看... Excel是一种电子表格格式,广泛用于数据处理和分析,而HTM则是一种用于创建网页的标记语言。虽然两

Java中Springboot集成Kafka实现消息发送和接收功能

《Java中Springboot集成Kafka实现消息发送和接收功能》Kafka是一个高吞吐量的分布式发布-订阅消息系统,主要用于处理大规模数据流,它由生产者、消费者、主题、分区和代理等组件构成,Ka... 目录一、Kafka 简介二、Kafka 功能三、POM依赖四、配置文件五、生产者六、消费者一、Kaf

使用Python实现在Word中添加或删除超链接

《使用Python实现在Word中添加或删除超链接》在Word文档中,超链接是一种将文本或图像连接到其他文档、网页或同一文档中不同部分的功能,本文将为大家介绍一下Python如何实现在Word中添加或... 在Word文档中,超链接是一种将文本或图像连接到其他文档、网页或同一文档中不同部分的功能。通过添加超

windos server2022里的DFS配置的实现

《windosserver2022里的DFS配置的实现》DFS是WindowsServer操作系统提供的一种功能,用于在多台服务器上集中管理共享文件夹和文件的分布式存储解决方案,本文就来介绍一下wi... 目录什么是DFS?优势:应用场景:DFS配置步骤什么是DFS?DFS指的是分布式文件系统(Distr

NFS实现多服务器文件的共享的方法步骤

《NFS实现多服务器文件的共享的方法步骤》NFS允许网络中的计算机之间共享资源,客户端可以透明地读写远端NFS服务器上的文件,本文就来介绍一下NFS实现多服务器文件的共享的方法步骤,感兴趣的可以了解一... 目录一、简介二、部署1、准备1、服务端和客户端:安装nfs-utils2、服务端:创建共享目录3、服

C#使用yield关键字实现提升迭代性能与效率

《C#使用yield关键字实现提升迭代性能与效率》yield关键字在C#中简化了数据迭代的方式,实现了按需生成数据,自动维护迭代状态,本文主要来聊聊如何使用yield关键字实现提升迭代性能与效率,感兴... 目录前言传统迭代和yield迭代方式对比yield延迟加载按需获取数据yield break显式示迭

Python实现高效地读写大型文件

《Python实现高效地读写大型文件》Python如何读写的是大型文件,有没有什么方法来提高效率呢,这篇文章就来和大家聊聊如何在Python中高效地读写大型文件,需要的可以了解下... 目录一、逐行读取大型文件二、分块读取大型文件三、使用 mmap 模块进行内存映射文件操作(适用于大文件)四、使用 pand

python实现pdf转word和excel的示例代码

《python实现pdf转word和excel的示例代码》本文主要介绍了python实现pdf转word和excel的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价... 目录一、引言二、python编程1,PDF转Word2,PDF转Excel三、前端页面效果展示总结一