tonic 1.3.3.dev3__tar.gz → 1.3.4.dev3__tar.gz

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/ChangeLog

@@ -1,6 +1,13 @@
 CHANGES
 =======
 
+* Added compatability for non-scalar targets
+* add ms\_to\_idx to DSEC data
+
+v1.3.3
+------
+
+* use numpy struct to unstruct function for timesurface generation
 * try/catch case where DAVIS dataset doesn't have optitrack targets
 * update plot\_animation fn
 * update plot\_animation fn

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tonic
-Version: 1.3.3.dev3
+Version: 1.3.4.dev3
 Summary: Neuromorphic datasets and transformations.
 Home-page: UNKNOWN
 Author: The Neuromorphs of Telluride

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/tonic/collation.py

@@ -48,7 +48,10 @@ class PadTensors:
                 )
             samples_output.append(sample)
             targets_output.append(target)
-        return (
-            torch.stack(samples_output, 0 if self.batch_first else 1),
-            torch.tensor(targets_output, device=target.device),
-        )
+        
+        samples_output = torch.stack(samples_output, 0 if self.batch_first else 1)
+        if len(targets_output[0].shape) > 1:
+            targets_output = torch.stack(targets_output, 0 if self.batch_first else -1) 
+        else:
+            targets_output = torch.tensor(targets_output, device=target.device)
+        return (samples_output, targets_output)

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/tonic/datasets/dsec.py

@@ -247,14 +247,16 @@ class DSEC(Dataset):
             full_base_folder = os.path.join(base_folder, data_name)
             if data_name in ["events_left", "events_right"]:
                 with h5py.File(full_base_folder + "/events.h5", "r") as file:
-                    data = make_structured_array(
+                    data = {}
+                    data[data_name] = make_structured_array(
                         file["events"]["x"][()],
                         file["events"]["y"][()],
                         file["events"]["t"][()],
                         file["events"]["p"][()],
                         dtype=self.dtype,
                     )
-                    data["t"] += file["t_offset"][()]
+                    data[data_name]["t"] += file["t_offset"][()]
+                    data["ms_to_idx"] = file["ms_to_idx"][()]
 
             elif "images" in data_name:
                 images_rectified_filenames = sorted(

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/tonic/functional/to_timesurface.py

@@ -1,6 +1,7 @@
 from typing import Tuple
 
 import numpy as np
+from numpy.lib import recfunctions as rfn
 
 from tonic.slicers import slice_events_by_time
 
@@ -41,7 +42,7 @@ def to_timesurface_numpy(
     start_t = event_slices[0][0][t_index]
     for i, slice in enumerate(event_slices):
         # structured to unstructured in order to access the indices
-        slice = slice.view((int, len(slice.dtype.names)))
+        slice = rfn.structured_to_unstructured(slice, dtype=int)
         indices = slice[:, [p_index, y_index, x_index]].T
         timestamps = slice[:, t_index]
         memory[tuple(indices)] = timestamps

{tonic-1.3.3.dev3 → tonic-1.3.4.dev3}/tonic.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tonic
-Version: 1.3.3.dev3
+Version: 1.3.4.dev3
 Summary: Neuromorphic datasets and transformations.
 Home-page: UNKNOWN
 Author: The Neuromorphs of Telluride

tonic-1.3.4.dev3/tonic.egg-info/pbr.json

@@ -0,0 +1 @@
+{"git_version": "7be656b", "is_release": false}

tonic-1.3.3.dev3/tonic.egg-info/pbr.json

@@ -1 +0,0 @@
-{"git_version": "142d3b4", "is_release": false}