Skip to content

Include Classification and Segmentation Heads

Latest
Compare
Choose a tag to compare
@ltriess ltriess released this 02 Jun 07:17
· 1 commit to main since this release
  • INTERFACE BREAK: rename attributes for feature and coordinate normalization
  • add implementation for classification head
  • add implementation for feature propagation module
  • add implementation for segmentation head
  • properly propagate training and mask tensors through the modules
  • enable usage of different feature normalization layers
  • add option to remove dropout layers
  • some bugfixes
  • update docstrings and readme