- 博客(5)
- 收藏
- 关注
RSS订阅原创 Onnx 导出swin transformer
torch 2.0 支持onnx 17，支持直接导出LayerNormalization。重新配置环境为torch2.0。1. 按照repo配置好swin transformer 环境。2. 在repo 目录下创建文件export.py.4. 简化Layer Normalization。运行以上程序回产生roll算子不支持的报错。然后在运行如上的export程序即可。3. 解决算子不支持的问题。导出结果用netron打开。
2023-11-13 08:52:30
427
1
原创 DCI-VTON-Virtual-Try-On
Repo:GitHub - bcmi/DCI-VTON-Virtual-Try-On: [ACM Multimedia 2023] Taming the Power of Diffusion Models for High-Quality Virtual Try-On with Appearance Flow.Paper:https://arxiv.org/pdf/2308.06101.pdf
2023-11-08 14:15:35
278
1
原创 ONNX导出失败
修改opset的版本 查看不支持的算子在新的opset中是否被支持 如果不考虑自己做定制plugin的话，也需要看看onnx-trt中这个算子是否被支持(*) ONNX是一种图结构，并不包含算子的实现，除非要跑onnx-runtime。我们关注的是onnx-trt的转换的支持。 看官方文档 替换pytorch中的算子组合 把某些算子替换为onnx可以识别的 在pytorch登记onnx中某些算子 有可能onnx中有支持，但此转换登记 替换修改o
2023-10-17 02:30:17
193
原创 ONNX构架剖析和ProtoBuf
ONNX（Open Neural Network Exchange Intermediate Representation）是一种神经网络的格式，采用Protobuf二进制形式进行序列化模型。Protobuf 全称叫做Protocal Buffer。是Google提出来的一套表示和序列化数据的机制。Protobuf会根据用于定义的数据结构来进行序列化存储同理，我们可以根据官方提供的数据结构信息，去修改或者创建onnx。Protobuf的编译以及序列化流程。
2023-10-10 02:02:14
378
1
原创 AI实战营开营仪式
这里是我们的repo，要求在github issue 上面提交作业。主讲人子豪，这些是我们的助教。我们班的助教林老师做红外检测。
2023-06-01 13:18:20
154
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人