From 9eb1b7b05acc92ec2b5e98f4f34c67dedcecb00c Mon Sep 17 00:00:00 2001 From: Chris Choy Date: Wed, 16 Oct 2019 15:58:49 -0700 Subject: [PATCH] fix #3, update readme on scannet split --- README.md | 11 +++++++---- lib/datasets/scannet.py | 8 ++++---- 2 files changed, 11 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index b3fac59..11da688 100644 --- a/README.md +++ b/README.md @@ -5,18 +5,21 @@ This repository contains the accompanying code for [4D-SpatioTemporal ConvNets: ## ScanNet Training -First, preprocess all scannet raw point cloud with the following command after you set the path correctly. +1. First, preprocess all scannet raw point cloud with the following command after you set the path correctly. ``` python -m lib.datasets.prepreocessing.scannet ``` -Then, train the scannet network with +2. Download the v2 official splits from [https://github.com/ScanNet/ScanNet/tree/master/Tasks/Benchmark](https://github.com/ScanNet/ScanNet/tree/master/Tasks/Benchmark) and save them to the scannet preprocessed root directory. + +3. Create `scannetv2_trainval.txt` by concatenating `scannetv2_train.txt` and `scannetv2_val.txt`. + +4. Train the scannet network with ``` -./scripts/train_scannet.sh 0 -default "--scannet_path /path/to/preprocessed/scannet" +export BATCH_SIZE=N; ./scripts/train_scannet.sh 0 -default "--scannet_path /path/to/preprocessed/scannet" ``` The first argument is the GPU id and the second argument is the path postfix and the last argument is the miscellaneous arguments. - diff --git a/lib/datasets/scannet.py b/lib/datasets/scannet.py index 735ce72..7051fc3 100644 --- a/lib/datasets/scannet.py +++ b/lib/datasets/scannet.py @@ -78,10 +78,10 @@ class ScannetVoxelizationDataset(VoxelizationDataset): # If trainval.txt does not exist, copy train.txt and add contents from val.txt DATA_PATH_FILE = { - DatasetPhase.Train: 'train.txt', - DatasetPhase.Val: 'val.txt', - DatasetPhase.Val: 'trainval.txt', - DatasetPhase.Test: 'test.txt' + DatasetPhase.Train: 'scannetv2_train.txt', + DatasetPhase.Val: 'scannetv2_val.txt', + DatasetPhase.TrainVal: 'scannetv2_trainval.txt', + DatasetPhase.Test: 'scannetv2_test.txt' } def __init__(self,