img_size 變更後error

[bryan0806]

trafficstars

我將config 的img_size 變更為384之後 先是發生 cuda memory 爆掉 RuntimeError: CUDA out of memory. Tried to allocate 82.00 MiB (GPU 0; 7.93 GiB total capacity; 6.47 GiB already allocated; 63.56 MiB free; 6.50 GiB reserved in total by PyTorch)

然後我變更batch size = 32之後發生下面error File "/home/alan/Downloads/movenet.pytorch/lib/loss/movenet_loss.py", line 304, in maxPointPth heatmap = heatmap*self.center_weight[:heatmap.shape[0],...] RuntimeError: The size of tensor a (96) must match the size of tensor b (48) at non-singleton dimension 3

請問 img_size 能否加大 還是只能用192 加大後哪邊需要更改呢 感謝

[fire717]

尺寸可以加大，但是不能直接修改尺寸，因为涉及到特征图的权重矩阵。 修改图像尺寸后，需要重新生成数据，然后修改两个特征图相乘的权重 可以参考这个issue的相关讨论：https://github.com/fire717/movenet.pytorch/issues/1