retinaface-pytorch icon indicating copy to clipboard operation
retinaface-pytorch copied to clipboard

随着训练,显存占用增大?

Open TryHard-LL opened this issue 3 years ago • 7 comments

哈喽,老师~ 请问您有遇到过随着训练epoch的增加,显存逐渐增大的问题嘛? 我更换了backbone后,会不断出现这样的情况~~

TryHard-LL avatar Nov 16 '22 04:11 TryHard-LL

一般重配conda环境可以解决

bubbliiiing avatar Nov 16 '22 14:11 bubbliiiing

老师,请问在自定义的DataGenerator中,get_random_data方法里 “对真实框进行调整”这部分是什么样的目的及原理,还有规则是什么呢?

TryHard-LL avatar Dec 13 '22 02:12 TryHard-LL

因为数据增强了,扭曲了,box也变了

bubbliiiing avatar Dec 13 '22 15:12 bubbliiiing

因为数据增强了,扭曲了,box也变了

嗯嗯,对于您说的扭曲变换了,box也需要变,就是以下这部分我稍有疑惑: “ box[:, [0, 2, 4, 6, 8, 10, 12]] = w - box[:, [2, 0, 6, 4, 8, 12, 10]] box[:, [5, 7, 9, 11, 13]] = box[:, [7, 5, 9, 13, 11]] ” 以上这部分是根据点来做对称变换嘛?(变换规则是类似左右眼对换嘛?)

TryHard-LL avatar Dec 14 '22 01:12 TryHard-LL

嗯?没听懂

bubbliiiing avatar Dec 18 '22 13:12 bubbliiiing

假如左眼的关键点的编号为1, 右眼的编号为2,那么在做扭曲变换时候,是不是应该是 landmarks[1, 2] = landmarks [2, 1],也就是这种变换是必须是对称变换,即左右眼的关键点信息进行对称的变换。 我阐述的清楚了嘛。。

TryHard-LL avatar Dec 19 '22 02:12 TryHard-LL

前面不是有个变换是flip吗

bubbliiiing avatar Dec 23 '22 15:12 bubbliiiing