人体解析语义分割转工程应用
阿瓜瓜瓜
这个作者很懒,什么都没留下…
展开
-
libtorch环境搭建、生成库文件、测试
Windows 配置 win10、GeForce 930M、cuda9.0、cudnn7.4.2 环境搭建 win下可以直接在官网下载编译好的libtorch(1.1),附一下libtorch安装以前版本的方法 https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.1.0.zip https:...原创 2020-03-13 10:45:26 · 3251 阅读 · 4 评论 -
模型转换
有两种将PyTorch模型转换为Torch脚本的方法。第一种是通过trace转换,第二种是通过script转换,trace比script简单,但只适合结构固定的网络模型,即forward中没有控制流的情况,因为trace只会保存运行时实际走的路径。如果forward函数中有控制流(IfElse、Switch、While、DynamicRNN、StaticRNN),需要用script方式实现。 这里...原创 2020-03-12 10:10:24 · 416 阅读 · 0 评论 -
模型准备
项目需完成的任务是人体的部位分割 ,需要判断人体部位类别,属于人体解析的范畴。由于后面需要转为c++可调用的模型,这里需要先准备好模型。 数据集 关于人体解析的数据集可以看这篇博客,本项目采用的数据集是Person-part数据集,以上博客关于Person-part的地址已经失效,我是从其他地方找到的数据集,但是数据集中只有标签图片,于是又自己下了VOC2010的原数据集,(附VOC数据集简介...原创 2020-03-12 08:51:31 · 657 阅读 · 0 评论