Spaces:
Running
on
Zero
Running
on
Zero
File size: 477 Bytes
4893ce0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
"""
S3DIS Dataset
Author: Xiaoyang Wu (xiaoyang.wu.cs@gmail.com)
Please cite our work if the code is helpful to you.
"""
import os
from .defaults import DefaultDataset
from .builder import DATASETS
@DATASETS.register_module()
class S3DISDataset(DefaultDataset):
def get_data_name(self, idx):
remain, room_name = os.path.split(self.data_list[idx % len(self.data_list)])
remain, area_name = os.path.split(remain)
return f"{area_name}-{room_name}"
|