emkarcinos / AITech-flats

Projekt magisterski AITech - klasyfikacja stylu wnętrz mieszkań i transfer stylu
Apache License 2.0
2 stars 0 forks source link

#7 Prototyp modelu #28

Closed emkarcinos closed 1 year ago

emkarcinos commented 1 year ago

Closes #7

Zrobiłem jupytera który załaduje wszystkie dane i wytrenuje ResNeta18.

Po treningu model jest wypluwany do experiment/models/<nazwa_modelu>/<data>.pt. Żeby z niego 'skorzystać' , zrobiłem prosty skrypt robiący inferencję, który bierze jako wejście plik z modelem i plik ze zdjęciem i wypluwa nam nazwę stylu.

python .\inference.py -m .\models\ResNet18\22-Dec-04-13-23.pt -i ..\data\images\test\Scandinavian\1.jpg

Scandinavian

Na razie nie trenowałem za bardzo tego modelu, myślę że dziś zapuszczę na noc na większych rozmiarach obrazów i na więcej iteracji, to zobaczymy jakie ciekawe rezultaty przyniesie.

Wybaczcie za noise z commitami z pylintem, bazowałem na dwóch branchach i nie chciałem już wycofywać tych commitów. Jak PR #27 nie przejdzie to się je wyrzuci

emkarcinos commented 1 year ago

Feedback od Andre