Make number of filter and SE block adjustable

ai-med / quickNAT_pytorch

PyTorch Implementation of QuickNAT and Bayesian QuickNAT, a fast brain MRI segmentation framework with segmentation Quality control using structure-wise uncertainty

MIT License

101 stars 36 forks source link

Make number of filter and SE block adjustable #22

Closed oezguensi closed 4 years ago

oezguensi commented 4 years ago

In the initialization of QuickNat the number of filter as well as the type of S&E is hardcoded.