SegFormer3D
SegFormer3D copied to clipboard
Added SegFormer3D arch compatible with HF Transformers structure
- Current modeling impl is compatible with HF format. The goal is to integrate our model to Transformers open model repo
- Added config script to be imported to the main model parameters for better scalability
- SegFormer3D handles all the shapes bug free. No need to be cube root shape
- Reformatted the code-base with
black - updated
.gitignoreby removing unnecessary lines.
TODO:
- [ ] check if the output shape is expected behavior
- [ ] check for any logical bug withing our
modeling_segformer3d.pyscript - [ ] train the supposed bug free model on all the data sets used in our paper