fix a horrible bug where the pixels were treated as signed values

bbawj · Mar 2, 2024 · 821023f · 821023f
1 parent db43ba0
commit 821023f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/common.h b/common.h
@@ -85,7 +85,9 @@ Mat init_train_set(char *img_data, char *label_data, int n_images) {
         for (int row = 0; row < IMG_HEIGHT; ++row) {
             for (int col = 0; col < IMG_WIDTH; ++col) {
                 float pixel =
-                    img_data[image * IMG_SIZE + row * IMG_WIDTH + col] / 255.0f;
+                    (unsigned char)
+                        img_data[image * IMG_SIZE + row * IMG_WIDTH + col] /
+                    255.0f;
                 MAT_AT(t, image, row * IMG_WIDTH + col) = pixel;
             }
         }