2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Attempted to fix image loading

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-19 15:18:10 +02:00
parent 5f9e8d52f3
commit 517bc51118
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/twomartens/masterthesis/data.py
View File

@ -25,8 +25,8 @@ Functions:
from typing import Callable, List, Mapping, Tuple, Any, Dict from typing import Callable, List, Mapping, Tuple, Any, Dict
from typing import Sequence from typing import Sequence
import cv2
import numpy as np import numpy as np
import scipy
import tensorflow as tf import tensorflow as tf
from scipy import ndimage from scipy import ndimage
@ -280,11 +280,12 @@ def prepare_scenenet_val(data_path: str, protobuf_path: str) -> Tuple[List[List[
file_names_instances_traj.append(instance_file) file_names_instances_traj.append(instance_file)
# load instance file # load instance file
instance_image = np.array(cv2.imread(instance_file)) instance_image = scipy.misc.imread(instance_file)
for instance_id in instances_traj: for instance_id in instances_traj:
instance_local = np.copy(instance_image) instance_local = np.copy(instance_image)
instance_local[instance_local != instance_id] = 0 instance_local[instance_local != instance_id] = 0
instance_local[instance_local == instance_id] = 1 instance_local[instance_local == instance_id] = 1
print(instance_local)
coordinates = ndimage.find_objects(instance_local)[0] coordinates = ndimage.find_objects(instance_local)[0]
x = coordinates[0] x = coordinates[0]
y = coordinates[1] y = coordinates[1]