Streamlined normalization setup/reload from checkpoint on models
parent
6f3fc629
No related branches found
No related tags found
Showing
- helpers/generate_histograms.py 1 addition, 1 deletionhelpers/generate_histograms.py
- src/mednet/engine/trainer.py 8 additions, 31 deletionssrc/mednet/engine/trainer.py
- src/mednet/models/classify/alexnet.py 4 additions, 28 deletionssrc/mednet/models/classify/alexnet.py
- src/mednet/models/classify/densenet.py 4 additions, 28 deletionssrc/mednet/models/classify/densenet.py
- src/mednet/models/detect/faster_rcnn.py 6 additions, 31 deletionssrc/mednet/models/detect/faster_rcnn.py
- src/mednet/models/model.py 32 additions, 13 deletionssrc/mednet/models/model.py
- src/mednet/models/segment/driu.py 6 additions, 34 deletionssrc/mednet/models/segment/driu.py
- src/mednet/models/segment/driu_bn.py 6 additions, 34 deletionssrc/mednet/models/segment/driu_bn.py
- src/mednet/models/segment/driu_od.py 6 additions, 34 deletionssrc/mednet/models/segment/driu_od.py
- src/mednet/models/segment/driu_pix.py 6 additions, 34 deletionssrc/mednet/models/segment/driu_pix.py
- src/mednet/models/segment/hed.py 6 additions, 35 deletionssrc/mednet/models/segment/hed.py
- src/mednet/models/segment/lwnet.py 2 additions, 0 deletionssrc/mednet/models/segment/lwnet.py
- src/mednet/models/segment/m2unet.py 6 additions, 6 deletionssrc/mednet/models/segment/m2unet.py
- src/mednet/models/segment/unet.py 6 additions, 34 deletionssrc/mednet/models/segment/unet.py
- src/mednet/scripts/train.py 19 additions, 1 deletionsrc/mednet/scripts/train.py
- tests/classify/test_cli.py 23 additions, 3 deletionstests/classify/test_cli.py
- tests/segment/test_cli.py 27 additions, 3 deletionstests/segment/test_cli.py
Loading
Please register or sign in to comment