运行模型时MEM不断增长的问题

[zongtaili]

您好，我在docker中应用您训练好的模型，使用
python detect.py --weights weights/best.pt

当docker内存限制为200M时，可以运行一张4MB的图像
但是同样的200M下，不能运行两张1MB的图像，显示out of memory

进一步发现，200M内存下，执行两次python detect.py --weights weights/best.pt，每次处理一张1MB图片也会OOM
说明程序存在额外的内存占用，导致内存的需求不断增长

[zongtaili]

kubectl logs detector
Fri Jun 11 13:54:12 UTC 2021
Using CPU
Model Summary: 140 layers, 7.28697e+06 parameters, 6.61683e+06 gradients
/bin/bash: line 1: 8 Killed python3.6 detect.py --weights weights/best.pt --save-txt
Fri Jun 11 13:54:37 UTC 2021
Using CPU
Model Summary: 140 layers, 7.28697e+06 parameters, 6.61683e+06 gradients
/bin/bash: line 1: 18 Killed python3.6 detect.py --weights weights/best.pt --save-txt
Fri Jun 11 13:54:58 UTC 2021