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

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