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仿真......希望能帮到你!

相关推荐
Nebula_g1 小时前
JavaSE高级:动态代理
java·开发语言·算法·javase
计算机编程指导师2 小时前
计算机大数据项目推荐:基于Hadoop+Django的人工智能社交媒体情绪数据分析可视化系统源码 毕业设计 选题推荐 毕设选题 数据分析 机器学习
大数据·hadoop·python·计算机·spark·毕业设计·社交媒体
青春逝如流水2 小时前
Java 线程与操作系统、硬件的关系:状态、映射与差异
java·开发语言·多线程·ai文章
谢亮_vipxieliang2 小时前
Go gRPC基础与实战
开发语言·后端·golang
Aurorar0rua2 小时前
CS50 x 2024 Notes Data structures - 08
c语言·开发语言·学习方法
高级程序源2 小时前
springboot荆州旅游推介平台24175-计算机课程设计、毕业设计
vue.js·spring boot·python·mysql·django·课程设计
企业数字化笔记2 小时前
新租户注册成功却不能用?开通状态、初始化恢复与可用性验收
开发语言·python
Guarding and trust2 小时前
项目拆解(二):模型输出是‘野‘的——三层兜底+重分析,把AI变成可靠零件
java·人工智能·python·大模型·fastapi·ollama·错误兜底
瑶山2 小时前
Windows + WSL2 Ubuntu + VS Code配置C++开发环境
开发语言·c++·windows·vscode·wsl
用户298698530142 小时前
Python 实战:将 HTML 内容保存为图像文件
后端·python·html