Klasifikace obrázků pomocí PyTorch a Windows ML

Poznámka:

Pro větší funkčnost lze PyTorch použít také s DirectML ve Windows.

V tomto kurzu se dozvíte, jak trénovat model neurální sítě klasifikace obrázků pomocí PyTorchu, exportovat model do formátu ONNX a nasadit ho do aplikace Windows Machine Learning spuštěné místně na vašem zařízení s Windows.

Vyžaduje se základní znalosti v programovacích jazycích Python a C#. Předchozí zkušenosti s strojovém učením jsou vhodnější, ale nevyžadují se.

Pokud chcete přejít rovnou k instalaci, přečtěte si téma Instalace PyTorchu.

Pokud jste už nastavili PyTorch, začněte proces učení modelu získáním dat.

Jakmile budete připraveni začít s daty, můžete začít trénovat model a pak ho převést do formátu ONNX.

Pokud máte model ONNX a chcete zjistit, jak vytvořit úplně novou aplikaci WinML, přejděte k nasazení modelu.

Poznámka:

Pokud chcete, můžete naklonovat úložiště ukázek služby Windows Machine Learning a spustit dokončený kód pro účely tohoto kurzu. Řešení trénování PyTorch najdete tady nebo dokončenou aplikaci Windows ML. Pokud používáte soubor PyTorch, před spuštěním souboru se ujistěte, že jste nastavili příslušný interpret PyTorch.

Scénář

V tomto kurzu vytvoříme aplikaci pro klasifikaci obrázků strojového učení, která se dá spustit na jakémkoli zařízení s Windows. Model se vytrénuje tak, aby rozpoznal typy vzorů, a klasifikuje 10 popisků obrázků z vybrané trénovací sady. 

Požadavky pro PyTorch – trénování modelu:

PyTorch je podporován v následujících distribucích Systému Windows:

• Windows 7 a vyšší. Doporučuje se Windows 10 nebo vyšší.
• Windows Server 2008 r2 a novější

Pokud chcete používat Pytorch ve Windows, musíte mít nainstalovaný Python 3.x. Python 2.x není podporovaný.

Požadavky pro nasazení aplikace Windows ML

K vytvoření a nasazení aplikace WinML budete potřebovat následující:

• Windows 10 verze 1809 (build 17763) nebo novější Číslo verze sestavení můžete zkontrolovat spuštěním winver příkazu (Windows logo key + R)Spustit .
• Windows SDK pro build 17763 nebo vyšší Tady můžete získat sadu SDK.
• Visual Studio 2017 verze 15.7 nebo novější Pokud místo toho používáte VS2017, doporučujeme používat Visual Studio 2019 a některé snímky obrazovky v tomto kurzu se můžou lišit. Visual Studio získáte tady.
• Rozšíření Windows ML Code Generator (mlgen) pro Visual Studio. Stáhněte si VS 2019 nebo VS 2017.
• Budete také muset povolit vývojářský režim na počítači.

Poznámka:

Rozhraní API pro Windows ML jsou integrovaná do nejnovějších verzí Windows 10 (1809 nebo novějších) a Windows Serveru 2019. Pokud je vaše cílová platforma starší verze Windows, můžete aplikaci WinML přenést do distribuovatelného balíčku NuGet (Windows 8.1 nebo vyšší).

Další kroky

Začneme instalací PyTorchu a konfigurací našeho prostředí.

Důležité

PyTorch, logo PyTorch a všechny související značky jsou ochranné známky společnosti Facebook, Inc.