diff --git a/contact_graspnet/data.py b/contact_graspnet/data.py index 7b9a55f..1308296 100644 --- a/contact_graspnet/data.py +++ b/contact_graspnet/data.py @@ -301,10 +301,11 @@ def load_available_input_data(p, K=None): """ segmap, rgb, depth, pc_full, pc_colors = None, None, None, None, None + cam_K = None if K is not None: if isinstance(K,str): - cam_K = eval(K) + K = eval(K) cam_K = np.array(K).reshape(3,3) if '.np' in p: