Closed lvoegtlin closed 1 year ago
UNet now has a swappable classifier. This makes working with it way easier, as we can easily fine-tune it onto a dataset with more or less classes.
pytest or python run.py
pytest
python run.py
Kudos, SonarCloud Quality Gate passed!
0 Bugs 0 Vulnerabilities 0 Security Hotspots 0 Code Smells
82.1% Coverage 0.0% Duplication
Description
UNet now has a swappable classifier. This makes working with it way easier, as we can easily fine-tune it onto a dataset with more or less classes.
How to Test/Run?
pytest
orpython run.py