diff --git a/aimodel/src/lib/dataset/parse_heightmap.py b/aimodel/src/lib/dataset/parse_heightmap.py index 24861d9..3005906 100644 --- a/aimodel/src/lib/dataset/parse_heightmap.py +++ b/aimodel/src/lib/dataset/parse_heightmap.py @@ -6,7 +6,7 @@ from ..io.readfile import readfile def parse_heightmap(filepath_heightmap): obj = json.loads(readfile(filepath_heightmap)) - result = tf.constant(obj.data) + result = tf.constant(obj["data"]) result = tf.transpose(result, [1,0]) # [ height, width ] → [ width, height ] return result \ No newline at end of file