• Ziyi Wu's avatar
    [Feature] Support PointNet++ Segmentor (#528) · 9d852f17
    Ziyi Wu authored
    * build BaseSegmentor for point sem seg
    
    * add encoder-decoder segmentor
    
    * update mmseg dependency
    
    * fix linting errors
    
    * warp predicted seg_mask in dict
    
    * add unit test
    
    * use build_model to wrap detector and segmentor
    
    * fix compatibility with mmseg
    
    * faster sliding inference
    
    * merge master
    
    * configs for training on ScanNet
    
    * fix CI errors
    
    * add comments & fix typos
    
    * hard-code class_weight into configs
    
    * fix logger bugs
    
    * update segmentor unit test
    
    * logger use mmdet3d
    
    * use eps to replace hard-coded 1e-3
    
    * add comments
    
    * replace np operation with torch code
    
    * add comments for class_weight
    
    * add comment for BaseSegmentor.simple_test
    
    * rewrite EncoderDecoder3D to avoid inheriting from mmseg
    9d852f17
encoder_decoder.py 17.7 KB