fix #3, update readme on scannet split

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.
-

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,