1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

Bài tập

Definování ztrátových funkcí pro RPN a R-CNN

Chystáš se trénovat model pro detekci objektů, který využívá komponenty RPN i R-CNN. Aby bylo možné model trénovat, je třeba pro každou z těchto komponent definovat ztrátovou funkci.

Vzpomeň si, že komponenta RPN klasifikuje, zda oblast obsahuje objekt, a předpovídá souřadnice ohraničujících rámečků pro navržené oblasti. Komponenta R-CNN zařazuje objekt do jedné z více tříd a zároveň předpovídá finální souřadnice ohraničujících rámečků.

torch a torch.nn jako nn jsou již naimportovány.

Hướng dẫn

100 XP
  • Definuj ztrátovou funkci pro klasifikaci v RPN a přiřaď ji do rpn_cls_criterion.
  • Definuj ztrátovou funkci pro regresi v RPN a přiřaď ji do rpn_reg_criterion.
  • Definuj ztrátovou funkci pro klasifikaci v R-CNN a přiřaď ji do rcnn_cls_criterion.
  • Definuj ztrátovou funkci pro regresi v R-CNN a přiřaď ji do rcnn_reg_criterion.