diff options
| author | Edoardo Pasca <edo.paskino@gmail.com> | 2019-05-09 12:41:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-05-09 12:41:24 +0100 |
| commit | 6b7a0783a37d8c3309e7acf65be8644d1c6e0164 (patch) | |
| tree | 739ea12ce57a5d1326739e9d4a946a0168488787 | |
| parent | b377d100af892052c86c21ebac3b3ab29918b080 (diff) | |
| download | framework-6b7a0783a37d8c3309e7acf65be8644d1c6e0164.tar.gz framework-6b7a0783a37d8c3309e7acf65be8644d1c6e0164.tar.bz2 framework-6b7a0783a37d8c3309e7acf65be8644d1c6e0164.tar.xz framework-6b7a0783a37d8c3309e7acf65be8644d1c6e0164.zip | |
set proper directory on creation
| -rwxr-xr-x | Wrappers/Python/ccpi/framework/TestData.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Wrappers/Python/ccpi/framework/TestData.py b/Wrappers/Python/ccpi/framework/TestData.py index 61ed4df..7e8362e 100755 --- a/Wrappers/Python/ccpi/framework/TestData.py +++ b/Wrappers/Python/ccpi/framework/TestData.py @@ -21,7 +21,7 @@ class TestData(object): def load(self, which, size=(512,512), scale=(0,1), **kwargs):
if which not in [TestData.BOAT, TestData.CAMERA, TestData.PEPPERS]:
raise ValueError('Unknown TestData {}.'.format(which))
- tmp = Image.open(os.path.join(data_dir, which))
+ tmp = Image.open(os.path.join(self.data_dir, which))
data = numpy.array(tmp.resize(size))
|