Skip to content

Instantly share code, notes, and snippets.

@ksob
Created February 5, 2019 18:32
Show Gist options
  • Save ksob/d19be1fa18f6628b0e6532bb520385c1 to your computer and use it in GitHub Desktop.
Save ksob/d19be1fa18f6628b0e6532bb520385c1 to your computer and use it in GitHub Desktop.
Jak uruchomić image classifier
1. Musisz zainstalować fast.ai, czyli bibliotekę python'a, najlepiej za pomocą anacond'y (wersja 4.6.2 lub wyższa).
(https://www.anaconda.com/distribution/#download-section)
Dalsze kroki dotyczą tylko instalacji za pomocą anacond'y.
2. Stwórz środowisko anacond'y na którym będzie instalowane fast.ai poleceniem:
$ conda create -n moje_srodowisko python=3
(moje_srodowisko to dowolna nazwa twojego środowiska)
3. Aktywuj środowisko :
$ source activate moje_srodowisko
4. Zainstaluj biblioteki torch
4.1. Jeśli nie posiadasz karty graficznej z CUDA w wersji co najmniej 7.0, lub posiadasz ją
ale nie zależy ci na szybkości wykonania skryptu, będziesz musiał zainstalować wersję bibliotek
operujących tylko na procesorze. Zrobisz to wykonując polecenie :
$ conda install -c pytorch pytorch-cpu torchvision-cpu
4.2. Jeśli posiadasz kartę graf. z CUDA w wersji co najmniej 7.0 i chcesz by classifier działał szybciej niż wersja CPU,
będziesz musiał zainstalować inne biblioteki, w tym celu przejrzyj dokumentację fastai (https://github.com/fastai/fastai).
Instalacja tych bibliotek będzie wyglądała mniej więcej tak:
$ conda install -c pytorch pytorch cuda100
Musisz jednak przejrzeć dokumentację by dowiedzieć się jaką wersję biblioteki cuda zainstalować.
4. Zainstaluj biblioteki fastai :
$ conda install -c fastai fastai
5. Wejdź do katalogu ze skryptem start.py, umieść pliki zdjęć w folderze 'zdjecia' i wykonaj polecenie:
$ python start.py
6. Do pliku classification_results.txt będą dopisywane na bieżąco wyniki
Side notes:
Tutorial bazuje na dokumentacji instalacji Fast AI:
https://docs.fast.ai/install.html
Skrypt czyta jedynie pliki jpg, jpeg, png, bmp i gif,
działa tylko na pierwszej warstwie katalogu 'zdjęcia', a nie rekurencynie.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment