1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Kreslení ohraničujícího rámečku

Máš připravené tensory obrázku i ohraničujícího rámečku – teď můžeš rámeček nakreslit přímo přes obrázek a vizuálně ověřit jeho přesnost.

torch, torchvision a torchvision.transforms jsou již naimportované. Obrázek byl převeden na tensor jako image_tensor. Souřadnice jsou uloženy v proměnných: x_min, y_min, x_max, y_max.

Pokyny

100 XP
  • Importuj draw_bounding_boxes z torchvision.utils.
  • Definuj ohraničující rámeček bbox jako seznam obsahující x_min, y_min, x_max a y_max.
  • Předej image_tensor a bbox_tensor funkci draw_bounding_boxes, aby se rámeček nakreslil přes obrázek, a výsledek ulož do img_bbox.
  • Převeď tensor img_bbox na obrázek.