search

PW22-SBN-01 / dataset_pipeline

0 stars 0 forks source link

Merged Iterator #3

Closed AdityaNG closed 2 years ago

AdityaNG commented 2 years ago

Create a class MergedDatasetIterator which takes AndroidDatasetIterator, PandaDatasetIterator (and eventually a DVRDatasetIterator) as arguments and produces an object that allows access to the intersection of all the datasets.

Precompute things like valid start time, end time, duration, avg frame rate, etc. Raise warnings when there are large gaps in the data (>1 second)

https://github.com/PW22-SBN-01/dataset_pipeline/blob/26c31afd799bc6823ed68191da66b3f379c99528/dataset_helper/dataset_iterators.py#L234-L295

Harsh188 commented 2 years ago

I can work on this

AdityaNG commented 2 years ago

Got some basic merged iterator functionality running

https://github.com/PW22-SBN-01/dataset_pipeline/commit/bd08f323d01a54033744648509934580aa379f14

AdityaNG commented 2 years ago

Made the MergedIterator take and instance of PandaDatasetIterator and an instance of AndroidDatasetIterator as argument. It computes the intersection and IOU of the two and allows the user to iteratre over the intersection

https://github.com/PW22-SBN-01/dataset_pipeline/blob/be8d44c0f1e67dc85982739b65ab148906f3a488/dataset_helper/dataset_visualizer.py#L270-L275