Python图像处理——基于CSRNet的人群密度检测系统(Pytorch框架)

(1)开发环境

Windows,Anaconda3,Pycharm,Python3.9,CUDA11.6,torch1.13.1

(2)数据集介绍

使用的数据集是shanghaiTech数据集,该数据包含part_A_final和part_B_final两部分。A部分训练集300张图片,测试集182张图片;B部分训练集400张图片,测试集316张图片。A,B两部分测试集和训练集打开都包含images和ground_truth两部分。A部分从互联网上随机抓取的图像,B部分取自上海大都市繁华街道的图像。这两个子集之间的人群密度差异很大,使得人群的准确估计比大多数现有数据集更具挑战性。

本文方法:本文将A和B两个数据集合并到一起进行训练,其中训练集700张图片,测试集498张图片。

(3)功能描述

📚 模型训练验证

运行train.py进行模型训练,设置100轮,经过约24小时训练结束,验证集loss曲线和验证集的MAE曲线如图所示。

🎮 模型测试

导入单张图片使用test.py进行验证,输出人数和密度图,如图所示。

🌐 Pyqt5界面

主界面运行main_ui.py,包括导入模型、导入图片、阈值设置、开始检测。

视频演示及代码获取:Python图像处理------基于CSRNet的人群密度检测系统(Pytorch框架)

最后:

小编会不定期发布相关设计内容包括但不限于如下内容:信号处理、通信仿真、算法设计、matlab appdesigner,gui设计、simulink仿真......希望能帮到你!

相关推荐
烂蜻蜓8 小时前
Flask入门教程(七):请求与响应——Web应用的核心数据流
前端·python·flask
布莱克6058 小时前
单例模式详解:什么是单例模式,它能解决哪些问题?
开发语言·qt·单例模式
web行路人8 小时前
AI 全栈学习之旅 - Week4:容器化部署与生产环境实战
python·学习
charlie1145141918 小时前
deque、list 与 forward_list:vector 之外的三个选择
开发语言·数据结构·c++·list·开源项目
春涧草茶8 小时前
慢就是快 - 10-2函数 特殊字面量 返回值
python
慢云智慧空间8 小时前
空间智能与计算机视觉的本质差异:不止于识别,更是空间决策能力
人工智能·python·计算机视觉
mayaairi8 小时前
JS DOM属性操作完全指南:内容、样式与自定义属性
开发语言·前端·javascript
avi91118 小时前
Threejs新版本后glb导出提示.x问题GLTFExporter,替代3dmax
开发语言·前端·javascript·threejs·glb·gltfexporter
qq762118229 小时前
windows Qt mingw 编译libzip
开发语言·windows·qt
曹牧9 小时前
Java:Java 数组转 List
java·开发语言·windows