- 博客(3)
- 收藏
- 关注
RSS订阅原创 tensorflow冻结模型为pb文件的各种方法
笔者最近因为工作需要将TensorFlow训练模型迁移到晟腾芯片平台上离线推理，所以需要将ckpt或者h5模型冻结成pb，再利用ATC模型转换工具将pb转为离线模型om文件，期间遇到一些问题和坑，总结一下，供大家参考。1.Tensorflow1.x训练好的模型Ckpt文件：DB_resnet_v1_50_adam_model.ckpt-16801.indexDB_resnet_v1_50_adam_model.ckpt-16801.data-00000-of-00001DB_resnet_v1_5
2020-12-05 11:30:08
4021
3
原创 昇腾Mindstudio官方样例黑白图片上色
本应用支持运行在Atlas 200 DK上，实现了对黑白图像自动上色的功能。黑白图像上色是利用卷积计算提取特征，然后用卷积进行分类，从而计算出每个灰度图片的色彩预期。其中处理的图片为LAB图片，L通道即为亮度通道，AB通道为色彩通道，黑白上色模型就是推理AB通道数据然后叠加到L通道上，生成彩色LAB的图片，再将图片转换为BGR格式，即可以观看。原理可以在华为资深布道师毛红朝老师的直播中了解： http://live.vhall.com/880314441官网的样例链接：https://gitee.c
2020-12-05 11:05:59
2283
原创 昇腾Mindstudio官方样例源码解读分享---Resnet50推理
作为初学者，最近刚接触Ascend200DK，按照官方的指导一步步搭建起了开发+运行环境，指导可以参考下面这个链接：https://gitee.com/lovingascend/quick_start/blob/master/Atlas200DK_separate_MD/environment.md运行的第一个样例就是Mindstudio自带的Resnet50网络，对晟腾AI处理器的强大性能尤为印象深刻。个人习惯会去研究一下源码，这里分享下心得体会，能力有限，水平一般，如果哪里写的不对的，还请指正。对于
2020-12-05 10:29:56
3138
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人