fix double free

This commit is contained in:
lz 2022-03-31 18:30:35 +08:00
parent 4de5f45663
commit 8edc885312
2 changed files with 3 additions and 1 deletions

View File

@ -40,7 +40,7 @@ inception_v3.tflite
# hiai_vad.tflite
# ide_label_base.tflite
# ide_label_retrained.tflite
ml_ei_headpose.tflite
#ml_ei_headpose.tflite
# ml_ei_landmark.tflite
mnist.tflite
mobilenet.tflite

View File

@ -234,10 +234,12 @@ int main(int argc, const char **argv) {
ret = ReadCalibData(argv[4], &calib_tensors, &calib_num);
if (ret != kMSStatusSuccess) {
MSModelDestroy(&model_handle);
return ret;
}
ret = CompareOutputs(outputs_handle, &calib_tensors, calib_num);
if (ret != kMSStatusSuccess) {
MSModelDestroy(&model_handle);
return ret;
}
FreeCalibTensors(&calib_tensors, calib_num);
}