1. Learn
    2. /
  2. คอร์ส
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

แบบฝึกหัด

Model pro binární klasifikaci

Jako praktik v oblasti deep learningu patří trénování modelů pro klasifikaci obrázků k tvým hlavním úkolům. Často se setkáš s binární klasifikací, při které je potřeba rozlišit mezi dvěma třídami. Abys zefektivnil/a svůj pracovní postup a zajistil/a znovupoužitelnost, rozhodl/a ses vytvořit šablonu CNN modelu pro binární klasifikaci obrázků, kterou budeš moct využít v budoucích projektech.

Balíčky torch a torch.nn jsou importovány jako nn. Všechny obrázky mají velikost 64×64 pixelů.

คำแนะนำ

100 XP
  • Vytvoř konvoluční vrstvu se 3 vstupními kanály, 16 výstupními kanály, velikostí kernelu 3, krokem 1 a paddingem 1.
  • Vytvoř plně propojenou vrstvu se vstupní velikostí 16×32×32 a počtem tříd rovným 1; zadej hodnoty pouze v uvedeném pořadí (16*32*32, 1).
  • Vytvoř aktivační funkci sigmoid.