Zan Klanecek 0dc1bfc2ff fixed paths to Marvin data | il y a 3 ans | |
---|---|---|
.. | ||
trained_models | il y a 3 ans | |
README.md | il y a 3 ans | |
datareader.py | il y a 3 ans | |
model.py | il y a 3 ans | |
preprocessing.py | il y a 3 ans | |
run.py | il y a 3 ans | |
test.py | il y a 3 ans | |
train.py | il y a 3 ans | |
visualize_convolutions.py | il y a 3 ans | |
visualize_filters.py | il y a 3 ans | |
visualize_saliency.py | il y a 3 ans |
Za branje/odpiranje slik - ne spreminjaj.
Za arhitekturo konvolucijske mreže - spremeni le funkciji init
in forward
.
Za učenje modela in hkratno preverjanje uspešnosti na validacijski množici. Po potrebi spremeni način vrednotenja modela (trenutno je končna verjetnost za pripadnost razredu 0/1 definirana kot povprečje verjetnosti 10 centralnih rezin).
Za testiranje modela na testni množici. Ne spreminjaj batch_size=10
in shuffle=False
.
Primer uporabe učenja in testiranja na Marvinu. Za lokalno uporabo spremeni main_path_to_data
.
Vizualizacija filtrov naučenega modela.
Za vizualizacijo rezultatov konvolucij vhodne slike na različnih globinah mreže.
Mapa izpostavljenosti oz. saliency map za doprinos k verjetnosti za hudo okužbo.