From 097712c0ea9d36ce0dd0cf89554ea0a2efbd4a56 Mon Sep 17 00:00:00 2001 From: hofee Date: Wed, 30 Oct 2024 15:58:32 +0000 Subject: [PATCH] global_only: ratio2 --- core/nbv_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/nbv_dataset.py b/core/nbv_dataset.py index ca9c0c7..36f4ed1 100644 --- a/core/nbv_dataset.py +++ b/core/nbv_dataset.py @@ -48,7 +48,7 @@ class NBVReconstructionDataset(BaseDataset): for line in f: scene_name = line.strip() scene_name_list.append(scene_name) - return scene_name_list + return scene_name_list[:10] def get_datalist(self): datalist = []