zws-2019

hi, I also encountered this problem. Have you solved it?

> 还有多图形状不一致的时候，需要resize到同一个形状才可以。我是修改了data_mix.py里Sample_dataset里对多图做了统一形状。 > > 但还是会在build_mlp.py里出错。然后我又把sub_image reshape的第一维改成cnt(单图是1，多图就是图片数量)，后来就正常运行起来了。 > > sub_img = img.reshape(cnt,3,H//336,336,W//336,336).permute(0,2,4,1,3,5).reshape(-1,3,336,336).contigious() > >  这样看起来是可以跑通 4khd模型的处理逻辑看起来不支持多图 比如这里只把第一个image_feature作为glb_img，如果我有多图，逻辑就会有问题 ``` for [h, w] in shapes: B_ = h*w glb_img = image_features[:1] ###...