Zan Klanecek a5eb890ba4 pretrained got changed to weights in torch 1.0+		1 år sedan
..
trained_models	fd17827f49 initial commit to git0 psuf	2 år sedan
README.md	fd17827f49 initial commit to git0 psuf	2 år sedan
datareader.py	fd17827f49 initial commit to git0 psuf	2 år sedan
model.py	a5eb890ba4 pretrained got changed to weights in torch 1.0+	1 år sedan
preprocessing.py	6cbe531938 added preprocessing script	2 år sedan
run.py	fd17827f49 initial commit to git0 psuf	2 år sedan
test.py	fd17827f49 initial commit to git0 psuf	2 år sedan
train.py	fd17827f49 initial commit to git0 psuf	2 år sedan
visualize_convolutions.py	0dc1bfc2ff fixed paths to Marvin data	2 år sedan
visualize_filters.py	0dc1bfc2ff fixed paths to Marvin data	2 år sedan
visualize_saliency.py	0dc1bfc2ff fixed paths to Marvin data	2 år sedan

Navodila

I.) Učenje in testiranje modelov

1. datareader.py

Za branje/odpiranje slik - ne spreminjaj.

2. model.py

Za arhitekturo konvolucijske mreže - spremeni le funkciji init in forward.

3. train.py

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).

4. test.py

Za testiranje modela na testni množici. Ne spreminjaj batch_size=10 in shuffle=False.

5. run.py

Primer uporabe učenja in testiranja na Marvinu. Za lokalno uporabo spremeni main_path_to_data.

II.) Interpretacija modelov

1. visualize_filters.py

Vizualizacija filtrov naučenega modela.

2. visualize_convolutions.py

Za vizualizacijo rezultatov konvolucij vhodne slike na različnih globinah mreže.

3. visualize_saliency.py

Mapa izpostavljenosti oz. saliency map za doprinos k verjetnosti za hudo okužbo.

README.md