From 92ab3a7a7c2564a78f858644adc1aab7c09cae64 Mon Sep 17 00:00:00 2001 From: Danila Rukhovich Date: Tue, 15 Jun 2021 12:34:25 +0300 Subject: [PATCH] fix indoor_converter --- tools/data_converter/indoor_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/data_converter/indoor_converter.py b/tools/data_converter/indoor_converter.py index 7654f75e86..b408bd017d 100644 --- a/tools/data_converter/indoor_converter.py +++ b/tools/data_converter/indoor_converter.py @@ -86,7 +86,7 @@ def create_indoor_info_file(data_path, # no need to generate for test set train_dataset.get_seg_infos() val_dataset.get_seg_infos() - else: + elif pkl_prefix == 's3dis': # S3DIS doesn't have a fixed train-val split # it has 6 areas instead, so we generate info file for each of them # in training, we will use dataset to wrap different areas