|
il y a 3 semaines | |
---|---|---|
.. | ||
trained_models | il y a 3 semaines | |
README.md | il y a 3 semaines | |
datareader.py | il y a 3 semaines | |
model.py | il y a 3 semaines | |
training.py | il y a 3 semaines |
Prepared to read .npy stored CT axial slices.
CNN model architecture for 512x512 axial CT slices.
Simple script to train (training set) and validate (validation set) your model.
Note: Works with torch 1.9.