-
- Downloads
implement part segmentation for Hengshuang
Showing
- .gitignore 4 additions, 4 deletions.gitignore
- README.md 39 additions, 26 deletionsREADME.md
- config/cls.yaml 2 additions, 2 deletionsconfig/cls.yaml
- config/model/Hengshuang.yaml 4 additions, 4 deletionsconfig/model/Hengshuang.yaml
- config/model/Menghao.yaml 1 addition, 1 deletionconfig/model/Menghao.yaml
- config/model/Nico.yaml 8 additions, 8 deletionsconfig/model/Nico.yaml
- config/partseg.yaml 21 additions, 0 deletionsconfig/partseg.yaml
- dataset.py 103 additions, 0 deletionsdataset.py
- models/Hengshuang/model.py 141 additions, 49 deletionsmodels/Hengshuang/model.py
- models/Hengshuang/transformer.py 44 additions, 44 deletionsmodels/Hengshuang/transformer.py
- models/Menghao/model.py 158 additions, 158 deletionsmodels/Menghao/model.py
- models/Nico/model.py 1 addition, 1 deletionmodels/Nico/model.py
- train_cls.py 2 additions, 2 deletionstrain_cls.py
- train_partseg.py 242 additions, 0 deletionstrain_partseg.py
Loading
Please register or sign in to comment