Merge pull request #9890 from dkurt:fix_torch_test

This commit is contained in:
Alexander Alekhin 2017-10-20 11:43:16 +00:00
commit 276459e57e
2 changed files with 2 additions and 2 deletions

View File

@ -169,7 +169,7 @@ TEST(Torch_Importer, run_deconv)
TEST(Torch_Importer, run_batch_norm)
{
runTorchNet("net_batch_norm");
runTorchNet("net_batch_norm", DNN_TARGET_CPU, "", false, true);
}
TEST(Torch_Importer, net_prelu)

View File

@ -24,7 +24,7 @@ if __name__ == '__main__':
cols = frame.shape[1]
rows = frame.shape[0]
net.setInput(dnn.blobFromImage(frame, 1.0, (inWidth, inHeight), (104.0, 177.0, 123.0), False))
net.setInput(dnn.blobFromImage(frame, 1.0, (inWidth, inHeight), (104.0, 177.0, 123.0), False, False))
detections = net.forward()
perf_stats = net.getPerfProfile()