From 5a96b37f16b3c29dd2509507b1cdfdff5d53c558 Mon Sep 17 00:00:00 2001 From: Cody Reading Date: Thu, 1 Apr 2021 12:04:45 -0400 Subject: [PATCH] Release 0.3.1 - Fix depth Map Config and Pillow version (#21) * Only use depth downsample factor if in config * Updated version of Pillow --- pcdet/datasets/dataset.py | 5 ++++- requirements.txt | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/pcdet/datasets/dataset.py b/pcdet/datasets/dataset.py index 63f5cc8..6da27c0 100644 --- a/pcdet/datasets/dataset.py +++ b/pcdet/datasets/dataset.py @@ -40,7 +40,10 @@ def __init__(self, dataset_cfg=None, class_names=None, training=True, root_path= self.grid_size = self.data_processor.grid_size self.voxel_size = self.data_processor.voxel_size - self.depth_downsample_factor = self.dataset_cfg.DEPTH_MAP.DOWNSAMPLE_FACTOR + + if "DEPTH_MAP" in self.dataset_cfg: + self.depth_downsample_factor = self.dataset_cfg.DEPTH_MAP.DOWNSAMPLE_FACTOR + self.total_epochs = 0 self._merge_all_iters_to_one_epoch = False diff --git a/requirements.txt b/requirements.txt index 2d57c2d..9f5b940 100644 --- a/requirements.txt +++ b/requirements.txt @@ -20,7 +20,7 @@ networkx==2.5 numba==0.52.0 numpy==1.20.1 oauthlib==3.1.0 -Pillow==8.1.0 +Pillow==8.1.2 protobuf==3.15.3 pyasn1==0.4.8 pyasn1-modules==0.2.8