scales annotations

mmp/a3/annotation.py

@@ -20,6 +20,12 @@ class AnnotationRect:
     def __array__(self) -> np.ndarray:
         return np.array([self.x1, self.y1, self.x2, self.y2])
 
+    def scale(self, factor: float):
+        self.x1 *= factor
+        self.x2 *= factor
+        self.y1 *= factor
+        self.y2 *= factor
+
     @staticmethod
     def fromarray(arr: np.ndarray):
         return AnnotationRect(arr[0], arr[1], arr[2], arr[3])

mmp/a4/dataset.py

@@ -29,7 +29,7 @@ class MMP_Dataset(torch.utils.data.Dataset):
         @param is_test: Whether this is the test set (True) or the validation/training set (False)
         """
         self.image_size = image_size
-        self.images = []
+        self.images: Sequence[Tuple[str, Sequence[AnnotationRect]]] = []
         self.anchor_grid = anchor_grid
         self.min_iou = min_iou
         self.is_test = is_test
@@ -72,8 +72,14 @@ class MMP_Dataset(torch.utils.data.Dataset):
         img_id = re.match(r".*(\/)([0-9]+)(\.[^\/]*$)", self.images[idx][0]).group(2)
         if self.is_test:
             return (img_tensor, torch.Tensor(), int(img_id))
+
+        scaled_annotations = []
+        for annotation in self.images[idx][1]:
+            annotation.scale(self.image_size / max(img.size[0], img.size[1]))
+            scaled_annotations.append(annotation)
+
         label_grid = get_label_grid(
-            anchor_grid=self.anchor_grid, gts=self.images[idx][1], min_iou=self.min_iou
+            anchor_grid=self.anchor_grid, gts=scaled_annotations, min_iou=self.min_iou
         )
         return (img_tensor, label_grid, int(img_id))
 

mmp/a4/document.tex

@@ -0,0 +1,70 @@
+\documentclass[11pt,a4paper]{article}
+
+% Language and encoding settings
+\usepackage[utf8]{inputenc}
+\usepackage[T1]{fontenc}
+\usepackage[english]{babel}
+
+% Page formatting
+\usepackage[left=1in, right=1in, top=1in, bottom=1in]{geometry}
+\usepackage{setspace}
+\onehalfspacing
+
+% Header/Footer
+\usepackage{fancyhdr}
+\pagestyle{fancy}
+\fancyhf{} % clear all header and footer fields
+\fancyhead[L]{\textbf{\course}}
+\fancyhead[C]{Assignment \assignmentnumber}
+\fancyhead[R]{\name}
+\fancyfoot[C]{\thepage}
+
+% Other packages
+\usepackage{enumitem}
+\usepackage{graphicx}
+
+% Custom commands for easy detail insertion
+\newcommand{\assignmentnumber}{04} % <-- CHANGE Assignment Number
+\newcommand{\name}{Simon Franken}      % <-- CHANGE YOUR NAME
+\newcommand{\course}{Multimedia Project WiSe 2526}  % <-- CHANGE COURSE NAME
+\newcommand{\duedate}{2025-11-12}  % <-- CHANGE DUE DATE
+
+% Title formatting
+\usepackage{titling}
+\pretitle{
+  \vspace*{2cm}
+  \begin{center}
+  \LARGE\bfseries
+}
+\posttitle{\par\end{center}\vspace{1cm}}
+
+\begin{document}
+
+\title{Assignment \assignmentnumber}
+\author{\name}
+\date{\duedate}
+
+\maketitle
+
+\begin{center}
+    \textbf{Course:} \course
+\end{center}
+\vspace{0.5cm}
+
+%------------------ START OF ASSIGNMENT -----------------------
+
+% Write your solutions below
+
+\section*{Exercise 4.2 Label Grid}
+\begin{enumerate}[label=\alph*)]
+    \setcounter{enumi}{2}
+    \item \begin{figure}[htp]
+              \centering
+              \includegraphics[width=4cm]{output.jpg}
+              \caption{output.txt}
+          \end{figure}
+\end{enumerate}
+
+%------------------ END OF ASSIGNMENT -----------------------
+
+\end{document}