Skip to content

Make the Flexible U-Net easily extensible for new backbones #5011

@binliunls

Description

@binliunls

Is your feature request related to a problem? Please describe.
Currently, the FlexibleUNet structure only supports the efficient-net series as the backbone and it is hard to extend to other network structures. A more extensible and convenient way to add more benchmark or user defined backbones will make it more flexible.

Describe the solution you'd like

  1. A base backbone/encoder class that defines interfaces like the number of output feature maps, a list of output feature map channels, string names of backbones and so on.

  2. A register who can add and update backbones informations to a dict.

With these two components, users can implement their own backbones class derived from the base backbone class and regist them to the backbone dict. Then Flexible U-Net will use this dict to generate backbones/encoders.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

Relationships

None yet

Development

No branches or pull requests

Issue actions