From 8e4b3c6c1799b5b2ede8025dd8b0be596a21b2aa Mon Sep 17 00:00:00 2001
From: Francesco Gatti <gattifrancesco@hotmail.t>
Date: Wed, 26 Jul 2017 01:46:25 -0900
Subject: [PATCH] download test data

---
 .gitignore            | 5 ++++-
 tests/build_models.sh | 7 +++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index f8a0d5a..6690668 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,7 @@
 build/
 .vscode/
 *.bin
-*.pyc
\ No newline at end of file
+*.pyc
+*.prototxt
+*.caffemodel
+*.h5
diff --git a/tests/build_models.sh b/tests/build_models.sh
index b3db436..c6b5fe2 100755
--- a/tests/build_models.sh
+++ b/tests/build_models.sh
@@ -1,4 +1,11 @@
 #!/bin/bash
+if [ "$1" == "download" ]; then
+    wget https://github.com/ceccocats/tkDNN/releases/download/testData/tkDNN_testwg.tar.gz --no-check-certificate
+    tar -xf tkDNN_testwg.tar.gz
+    rm tkDNN_testwg.tar.gz
+    exit
+fi
+
 echo "build test Model"
 cd test
 python test_model.py