统一的人群计数训练框架(PyTorch)——基于主流的密度图模型训练框架

视频讲解1:Bilibili视频讲解

视频讲解2:https://www.douyin.com/video/7583893329385999667?count=10&cursor=0&enter_method=post&modeFrom=userPost&previous_page=personal_homepage&secUid=MS4wLjABAAAA0NVS_BfnZjuBUqHzrh-1oSxoNxExvuesrznu1Wu4-fc

代码下载:https://github.com/KeepTryingTo/CrowdCounting-framework-PyTorch

本项目主要是实现了一个人群计数框架,学习者可以直接一键使用该框架生成密度图数据集,并且该框架也给出了一些常见的损失函数实现,只需要直接加入即可使用。该框架最大的亮点是,学习者不需要去改变任何代码即可训练模型,如果学习者想要加入自己的模型,直接实现即可嵌入到该训练框架中。该框架帮助学习者不需要去关注除了核心部分之外的其他代码,帮助学习者更专心的相关算法。 并且还实现了采用分布式框架训练模型。其中有关该项目参考的其他开源代码已经在项目末尾给出了链接。

支持的功能

√\] 单GPU,CPU或者多GPU分布式训练 \[√\] 自定义标注框数据集生成密度图 \[√\] 一键生成常见数据集的密度图之后即可进行训练,比如ShangHai_partA, ShangHai_partB,NWPU,QNRF,UCF_CC_50,JHUC-CROWD++ \[√\] 提供了常用的默认backbone网络模型 \[√\] 提供了常用的损失函数 \[√\] 日志记录以及模型保存(保存最好结果模型) \[√\] 模型参数量以及flops统计 \[√\] 绘制结果MAE训练曲线 \[√\] 可视化真实的密度图npy以及可视化模型预测密度图 \[√\] 提供了常用的学习率调度器 \[√\] 统计GPU的使用情况 更多使用说明请看: [https://github.com/KeepTryingTo/CrowdCounting-framework-PyTorch/blob/main/zh_README.md](https://github.com/KeepTryingTo/CrowdCounting-framework-PyTorch/blob/main/zh_README.md "https://github.com/KeepTryingTo/CrowdCounting-framework-PyTorch/blob/main/zh_README.md") 参考链接 [人群计数中常用数据集的总结以及使用方式(Python/PyTorch)](https://mydreamambitious.blog.csdn.net/article/details/147144537?spm=1011.2415.3001.5331 "人群计数中常用数据集的总结以及使用方式(Python/PyTorch)") [论文CrowdCLIP(基于CLIP的无监督人群计数模型)详解(PyTorch,Pytorch_Lighting)](https://mydreamambitious.blog.csdn.net/article/details/147851355?spm=1011.2415.3001.5331 "论文CrowdCLIP(基于CLIP的无监督人群计数模型)详解(PyTorch,Pytorch_Lighting)") [论文The Effectiveness of a Simplified Model Structure for Crowd Counting(FFNet)详解](https://mydreamambitious.blog.csdn.net/article/details/141355068?spm=1011.2415.3001.5331 "论文The Effectiveness of a Simplified Model Structure for Crowd Counting(FFNet)详解") [论文Domain-General Crowd Counting in Unseen Scenarios(DCCUS)详解以及对应代码详解](https://mydreamambitious.blog.csdn.net/article/details/142730047?spm=1011.2415.3001.5331 "论文Domain-General Crowd Counting in Unseen Scenarios(DCCUS)详解以及对应代码详解") [论文Distribution Matching for Crowd Counting详解](https://mydreamambitious.blog.csdn.net/article/details/143133438?spm=1011.2415.3001.5331 "论文Distribution Matching for Crowd Counting详解") [论文Distribution Matching for Crowd Counting中人群统计损失(C Loss),最优化传输损失(OT Loss)以及总的变化损失(TV Loss)](https://mydreamambitious.blog.csdn.net/article/details/143219789?spm=1011.2415.3001.5331 "论文Distribution Matching for Crowd Counting中人群统计损失(C Loss),最优化传输损失(OT Loss)以及总的变化损失(TV Loss)") [论文CLIP-EBC(基于CLIP的人群统计模型)详解](https://mydreamambitious.blog.csdn.net/article/details/147819012?spm=1011.2415.3001.5331 "论文CLIP-EBC(基于CLIP的人群统计模型)详解") [论文CLIP-Count(基于文本指导的零样本目标计数)详解(PyTorch)](https://mydreamambitious.blog.csdn.net/article/details/147873916?spm=1011.2415.3001.5331 "论文CLIP-Count(基于文本指导的零样本目标计数)详解(PyTorch)") https://github.com/cvlab-stonybrook/DM-Count https://github.com/gjy3035/Awesome-Crowd-Counting https://github.com/gjy3035/C-3-Framework https://github.com/ZPDu/Domain-general-Crowd-Counting-in-Unseen-Scenarios https://github.com/openai/CLIP https://mydreamambitious.blog.csdn.net/article/details/142730047?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/147144537?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/143133438?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/143219789?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/144692297?spm=1011.2415.3001.5331 https://github.com/erdongsanshi/FFNet https://mydreamambitious.blog.csdn.net/article/details/141355068?spm=1011.2415.3001.5331 https://github.com/KeepTryingTo/PyTorch-DeepLearning-Visual-LLM https://github.com/KeepTryingTo https://blog.csdn.net/Keep_Trying_Go/article/details/154913403 https://mydreamambitious.blog.csdn.net/article/details/154789450?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/154443638?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/148124251?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/148300284?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/148227727?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/147873916?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/147851355?spm=1011.2415.3001.5331 https://mydreamambitious.blog.csdn.net/article/details/147819012?spm=1011.2415.3001.5331

相关推荐
沅_Yuan12 小时前
基于核密度估计的BP-KDE多输入单输出回归模型【MATLAB】
机器学习·matlab·回归·bp·核密度估计·kde
User_芊芊君子12 小时前
全能远控,性能为王:UU远程深度测评与行业横评
人工智能·dubbo·测评
进击的荆棘12 小时前
递归、搜索与回溯——递归
算法·leetcode·递归
刀法如飞12 小时前
关于AI的三个核心问题——工具、认知与产业的再思考
人工智能·aigc·ai编程
2301_8227032013 小时前
鸿蒙Flutter第三方库FlutterUnit组件百科适配——具体示例还原演示1
算法·flutter·华为·harmonyos·鸿蒙
前端不太难13 小时前
一天做出:鸿蒙 + AI 游戏 Demo
人工智能·游戏·harmonyos
木斯佳16 小时前
HarmonyOS 6实战:AI Action富媒体卡片迭代——实现快照分享
人工智能·harmonyos·媒体
芝士爱知识a16 小时前
2026高含金量写作类国际竞赛汇总与测评
大数据·人工智能·国际竞赛·写作类国际竞赛·写作类比赛推荐·cwa·国际写作比赛推荐
ZhengEnCi17 小时前
M3-markconv库找不到wkhtmltopdf问题
python
华农DrLai19 小时前
什么是LLM做推荐的三种范式?Prompt-based、Embedding-based、Fine-tuning深度解析
人工智能·深度学习·prompt·transformer·知识图谱·embedding