
Pytorch
文章平均质量分 72
稳健的不高冷的强哥
这个作者很懒,什么都没留下…
展开
-
使用C++调用PyTorch模型的弯弯绕绕,推荐LibTorch加载,C++处理
本人莎莎的以为要重写PyTorch模型网络架构,于是使用C++的LibTorch和gpt重写了一下,结果一致卡在load不上,非常奇怪,于是检查两个网络的架构,其实非常类似,忘记截图载下来了,就下面一张图。.pth保存的是模型参数,通过字符字典进行保存,在加载该类文件时应该先实例化一个具体的模型,然后对新建立的空模型,进行参数赋予。cv 2,numpy,pillow等包都是如此,统一conda,或者统一使用pip,推荐使用pip。需要重写图像处理的一些代码,保持和Python一致,输入模型进行预测。原创 2024-08-19 12:27:59 · 1362 阅读 · 0 评论 -
基于Pytorch的图像分类总结:Swin Transformer
Swin Transformer 论文详解及程序解读 - 知乎 (zhihu.com)Swin Transformer 论文详解及程序解读 - 知乎 (zhihu.com)机器学习_详解 Swin Transformer (SwinT)_何处闻韶的博客-优快云博客_swin transformerSwin Transformer: Hierarchical Vision Transformer using Shifted Windows (arxiv.org)Swin-Transformer网络结构详解_原创 2022-06-21 13:45:14 · 1627 阅读 · 0 评论