Zan Klanecek 6cbe531938 added preprocessing script 3 년 전
..
trained_models fd17827f49 initial commit to git0 psuf 3 년 전
README.md fd17827f49 initial commit to git0 psuf 3 년 전
datareader.py fd17827f49 initial commit to git0 psuf 3 년 전
model.py fd17827f49 initial commit to git0 psuf 3 년 전
preprocessing.py 6cbe531938 added preprocessing script 3 년 전
run.py fd17827f49 initial commit to git0 psuf 3 년 전
test.py fd17827f49 initial commit to git0 psuf 3 년 전
train.py fd17827f49 initial commit to git0 psuf 3 년 전

README.md

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.