模型转换
qq_41669468
这个作者很懒,什么都没留下…
展开
-
pb模型转tflite(fp16, uint8)
@[TOC](pb模型转tflite(fp16, uint8))pb模型转tflitedef convert_pb_into_tflite_file(): # create a session input_tensor_name=["pb网络输入名"] input_tensor_shape={"pb网络输入名":[x,x,x,x]} #输入shape out_tensor_name = ["pb网络输出名"] converter = tf.li原创 2021-09-06 19:04:37 · 849 阅读 · 0 评论 -
模型转换pytorch->onnx->saved_model->tflite
目前的很多模型使用pytorch框架进行训练,移植到android时,需要将模型进行转换:ncnn, tflite等()原创 2020-12-07 17:29:13 · 2587 阅读 · 3 评论 -
YOLOv5转NCNN过程
官网提供的yolov5代码中已经能实现pt转onnx,但为了在android端使用 需要将onnx转ncnn,由于yolov5的部分网络层并没有在ncnn代码库中封装,直接使用ncnn中的onnx2ncnn会存在一些不支持的问题。该博客就是借他人经验,将实现的过程重新整理了一遍。便于向我一样的新手在初次使用时快速上手。实现步骤1, 安装onnx-simplifier1.1: pip install onnx-simplifier (具体该工具的作用自行百度)1.2: python -m onnxs原创 2020-12-02 17:13:42 · 6332 阅读 · 7 评论