我正在尝试使用 javascript 来试验 tensorflow 。我想在图像文件夹上运行posenet 模型。通过谷歌搜索,我编写了下面的 HTML。当我运行下面显示的 HTML 时,系统很快就会耗尽内存,确实存在内存泄漏。我不知道内存泄漏是在哪里以及如何发生的。社区可以解释一下内存泄漏是如何发生的以及如何解决它
<html>
<head>
<script src="https://unpkg.com/@tensorflow/tfjs"></script>
<script src="https://unpkg.com/@tensorflow-models/posenet"</script>
</head>
<body>
<img id='faceImg' src='1.0.jpg'/>
<script>
var i = 1;
async function makeTensors(){
var psData;
var imageScaleFactor = 0.5;
var outputStride = 16;
var flipHorizontal = false;
var iterator;
var imageElements = document.getElementById('faceImg');
console.log(imageElements);
let result = await posenet.load().then(function(net){
return net.estimateSinglePose(imageElements, imageScaleFactor, flipHorizontal, outputStride)
}).then(function(pose){
console.log(pose);
psData = pose;
})
var im_tensor_pos = []
for (iter = 0; iter < 17; iter++) {
im_tensor_pos.push(psData.keypoints[iter].position.x);
im_tensor_pos.push(psData.keypoints[iter].position.y);
}
for (iter = 0; iter < 17; iter++) {
im_tensor_pos.push(psData.keypoints[iter].score);
}
im_tensor_pos.push(psData.score);
// tensor_vals.push(im_tensor_pos);
psData = null;
delete result;
// console.log(tensor_vals);
}
async function test(){
for (i = 1; i < 270; i++) {
var ImageSrc = 'results/' + i + ".0.jpg";
console.log(ImageSrc)
var imageElements = document.getElementById('faceImg');
imageElements.src = ImageSrc;
result = await makeTensors();
delete result;
}
}
test().then(console.log('done'));
</script>
</body>
</html>
这也是我第一次尝试js。如果您对代码也有想法,请告诉我。
最佳答案
我遇到了完全相同的问题,但 .dispose
对我不起作用。
不知道您是否已经修复了它,但为了将来的引用,这有效:
tf.engine().startScope()
// do the Tensorflow / Posenet stuff
tf.engine().endScope()
完成后这将破坏张量
关于javascript - 在tensorflow JavaScript中使用posenet时出现内存泄漏,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55675185/