2024年8月9日
手写Swin Transformer
摘要： 1.如何基于图片生成patch_embedding? 方法一： - 基于pytorch unfold的API来将图片进行分块，也就是模仿卷积的思路，设置kernel_size=stride=patch_size,得到分块后的图片 - 得到格式为[bs, num_patch, patch_depth] 阅读全文
posted @ 2024-08-09 15:21 Trkly 阅读(46) 评论(0) 推荐(0)