Pedestron 开源项目使用教程
2024-09-20 23:56:45作者:薛曦旖Francesca
1. 项目介绍
Pedestron 是一个基于 MMdetection 的仓库,专注于行人检测研究的进展。该项目提供了一系列检测器,包括通用检测器和专门用于行人检测的检测器,用于训练和测试。此外,Pedestron 还提供了预训练模型和多个行人检测数据集上的基准测试。
2. 项目快速启动
安装
首先,克隆项目仓库并安装依赖项:
git clone https://github.com/hasanirtiza/Pedestron.git
cd Pedestron
根据 INSTALL.md 文件中的说明安装依赖项。
运行演示
使用预训练模型对图像进行检测:
python tools/demo.py configs/elephant/cityperson/cascade_hrnet.py /models_pretrained/epoch_5.pth demo/ result_demo/
训练
使用单个 GPU 进行训练:
python tools/train.py configs/elephant/cityperson/cascade_hrnet.py
使用多个 GPU 进行训练:
./tools/dist_train.sh configs/elephant/cityperson/cascade_hrnet.py 7
测试
使用单个 GPU 进行测试:
python tools/test_city_person.py configs/elephant/cityperson/cascade_hrnet.py /models_pretrained/epoch_5.pth --out result_citypersons.json --mean_teacher
使用多个 GPU 进行测试:
./tools/dist_test.sh configs/elephant/crowdhuman/cascade_hrnet.py /models_pretrained/epoch_19.pth 8 --out CrowdHuman12.pkl --eval bbox
3. 应用案例和最佳实践
应用案例
Pedestron 可以应用于自动驾驶、智能监控等领域,用于行人检测和行人行为分析。例如,在自动驾驶系统中,Pedestron 可以帮助车辆识别和跟踪行人,从而提高驾驶安全性。
最佳实践
- 数据集准备:根据
Datasets-PreProcessing.md文件中的说明准备数据集。 - 模型选择:根据具体需求选择合适的检测器和骨干网络。
- 超参数调优:通过调整训练参数和数据增强策略,优化模型性能。
4. 典型生态项目
- MMdetection:Pedestron 基于 MMdetection 框架,MMdetection 是一个强大的目标检测工具箱,支持多种检测器和骨干网络。
- Detectron2:Facebook AI Research 推出的目标检测框架,与 Pedestron 类似,支持多种检测任务。
- TensorFlow Object Detection API:Google 推出的目标检测框架,支持多种预训练模型和数据集。
通过结合这些生态项目,可以进一步扩展 Pedestron 的功能和应用场景。
登录后查看全文
热门项目推荐
相关项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0172
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook096
Step-3.7-FlashStep-3.7-Flash是一个拥有 1980 亿参数的稀疏混合专家(MoE)视觉语言模型,由 1960 亿参数的语言主干网络和 18 亿参数的视觉编码器组合而成,具备原生图像理解能力。Python00
BitCPM-CANN-8BBitCPM-CANN 是首个基于华为昇腾 NPU 原生构建的端到端 1.58 位(三值化)大语言模型训练系统。该系统将量化感知训练(QAT)集成到 Megatron-LM 框架中,并结合 MindSpeed 加速,覆盖了从自定义三值算子到基于昇腾 910B 的分布式并行训练的完整训练栈。Python00
MiniCPM5-1BMiniCPM5-1B,这是 MiniCPM5 系列的首款模型。它是一个专为端侧、本地部署和资源受限场景打造的 10 亿参数密集型 Transformer 模型,达到了 10 亿参数级开源模型的 SOTA 水平Jinja00
skillhubopenJiuwen 生态的 Skill 托管与分发开源方案,支持自建与可选 ClawHub 兼容。Python0239
项目优选
收起
deepin linux kernel
C
32
16
暂无描述
Dockerfile
749
4.87 K
Claude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed.
Get Started
Rust
1.56 K
172
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
840
1.83 K
Ascend Extension for PyTorch
Python
688
832
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
222
96
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
451
418
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.02 K
1.04 K
暂无简介
Dart
999
258
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
642
1.27 K