Transformer
ViT(TransReID)模型各阶段形状
已阅: 119 之前看TransReID代码时的记录,他们代码写的很全,训练测试都有。 这段时间在这个基础上改了点代码,暂时没出现模型理解错的地方。 ViT外的改进没有记录。 有错误或疑问请留言,谢谢。代码TransReID-GitHub 下面是TransReID里的ViT结构,不包括TransReID的改进。 代码简写了。参数设置: img_size = [H, W, C] = [224, 224, 3] # 图片尺寸,代码中的img_size为[H, W] patch_size = [16, 16] # patc……
(◕ܫ◕)
评论于 MMDet实例分割loss_rpn_bbox为nan但其它loss正常的解决