bubbliiiing/retinaface-pytorch

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

TryHard-LL opened this issue · 7 comments

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

一般重配conda环境可以解决

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

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

因为数据增强了,扭曲了,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]]

以上这部分是根据点来做对称变换嘛?(变换规则是类似左右眼对换嘛?)

嗯?没听懂

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

前面不是有个变换是flip吗