基于 Nvidia Jetson Nano 训练 SSD(单发多框检测器)用于人脸检测
计算机视觉与模式识别
2021-05-31 v1 人工智能
分布式、并行与集群计算
图像与视频处理
摘要
在本项目中,我们使用了计算机视觉算法 SSD(Single Shot detector,单发多框检测器)计算机视觉算法,并使用由 139 张图片组成的数据集对该算法进行了训练。图像使用 Intel CVAT(Computer Vision Annotation Tool,计算机视觉标注工具)进行标注。我们训练该模型用于人脸检测。我们已将训练好的模型与软件部署在 Nvidia Jetson Nano 开发套件中。模型代码使用 PyTorch 深度学习框架编写。所用编程语言为 Python。
引用
@article{arxiv.2105.13906,
title = {Training of SSD(Single Shot Detector) for Facial Detection using Nvidia Jetson Nano},
author = {Saif Ur Rehman and Muhammad Rashid Razzaq and Muhammad Hadi Hussian},
journal= {arXiv preprint arXiv:2105.13906},
year = {2021}
}
备注
7 Pages, 7 figures