From 10153f26344433e0ebf6f6d50fee47688e5ab427 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 19 Dec 2025 10:28:29 +0100 Subject: [PATCH] - fixed notebook - added .gitignore --- .gitignore | 2 + nn_image_classification.ipynb | 171 ++++++++-------------------------- 2 files changed, 39 insertions(+), 134 deletions(-) create mode 100644 .gitignore diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..f5a980d --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.ipynb_checkpoints/ +data diff --git a/nn_image_classification.ipynb b/nn_image_classification.ipynb index bc83125..ae6b11d 100644 --- a/nn_image_classification.ipynb +++ b/nn_image_classification.ipynb @@ -2,7 +2,7 @@ "cells": [ { "cell_type": "code", - "execution_count": 17, + "execution_count": 1, "id": "2e9496e9-ec07-445f-b29d-fc44aabb377d", "metadata": {}, "outputs": [], @@ -19,29 +19,31 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 2, "id": "9b17757b-76c5-4562-a2be-e3a50201c9ee", "metadata": {}, "outputs": [], "source": [ - "transform = transforms.Compose([transforms.ToTensor(), import numpy as np\n", - "from PIL import Image\n", - "import torch\n", - "import torch.nn as nn\n", - "import torch.nn.functional as F\n", - "import torch.optim as opim\n", - "import torchvision\n", - "import torchvision.transform as transform\n", - "\n", + "transform = transforms.Compose([\n", + " transforms.ToTensor(),\n", + " transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) \n", "]) " ] }, { "cell_type": "code", - "execution_count": 9, + "execution_count": null, "id": "5b898116-e9a8-429d-8491-67db5a65d358", "metadata": {}, - "outputs": [], + "outputs": [ + { + "name": "stderr", + "output_type": "stream", + "text": [ + "51.0%" + ] + } + ], "source": [ "train_data = torchvision.datasets.CIFAR10(root='./data', train=True, transform=transform, download=True)\n", "test_data = torchvision.datasets.CIFAR10(root='./data', train=False, transform=transform, download=True)\n", @@ -52,7 +54,7 @@ }, { "cell_type": "code", - "execution_count": 10, + "execution_count": null, "id": "49712fae-dfe6-4b6f-92ea-2e97a965c15c", "metadata": {}, "outputs": [], @@ -62,28 +64,17 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": null, "id": "d783e1a5-3410-4992-88a6-230dd2ec85dd", "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "torch.Size([3, 32, 32])" - ] - }, - "execution_count": 11, - "metadata": {}, - "output_type": "execute_result" - } - ], + "outputs": [], "source": [ "image.size()" ] }, { "cell_type": "code", - "execution_count": 12, + "execution_count": null, "id": "8ef21399-f2f2-46f4-8268-f60f36bbb533", "metadata": {}, "outputs": [], @@ -120,7 +111,7 @@ }, { "cell_type": "code", - "execution_count": 18, + "execution_count": null, "id": "f48dd819-9196-4aa5-92c4-ddcc644c6aa9", "metadata": {}, "outputs": [], @@ -132,77 +123,10 @@ }, { "cell_type": "code", - "execution_count": 19, + "execution_count": null, "id": "3a91ce0c-e71d-4bfb-b945-47edfc8ca8d6", "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Training epoch 0...\n", - "Loss: 2.1758\n", - "Training epoch 1...\n", - "Loss: 1.7324\n", - "Training epoch 2...\n", - "Loss: 1.5096\n", - "Training epoch 3...\n", - "Loss: 1.3996\n", - "Training epoch 4...\n", - "Loss: 1.3065\n", - "Training epoch 5...\n", - "Loss: 1.2286\n", - "Training epoch 6...\n", - "Loss: 1.1649\n", - "Training epoch 7...\n", - "Loss: 1.1101\n", - "Training epoch 8...\n", - "Loss: 1.0568\n", - "Training epoch 9...\n", - "Loss: 1.0140\n", - "Training epoch 10...\n", - "Loss: 0.9719\n", - "Training epoch 11...\n", - "Loss: 0.9321\n", - "Training epoch 12...\n", - "Loss: 0.8969\n", - "Training epoch 13...\n", - "Loss: 0.8649\n", - "Training epoch 14...\n", - "Loss: 0.8326\n", - "Training epoch 15...\n", - "Loss: 0.8014\n", - "Training epoch 16...\n", - "Loss: 0.7764\n", - "Training epoch 17...\n", - "Loss: 0.7449\n", - "Training epoch 18...\n", - "Loss: 0.7199\n", - "Training epoch 19...\n", - "Loss: 0.6991\n", - "Training epoch 20...\n", - "Loss: 0.6700\n", - "Training epoch 21...\n", - "Loss: 0.6438\n", - "Training epoch 22...\n", - "Loss: 0.6243\n", - "Training epoch 23...\n", - "Loss: 0.6051\n", - "Training epoch 24...\n", - "Loss: 0.5802\n", - "Training epoch 25...\n", - "Loss: 0.5609\n", - "Training epoch 26...\n", - "Loss: 0.5381\n", - "Training epoch 27...\n", - "Loss: 0.5162\n", - "Training epoch 28...\n", - "Loss: 0.4949\n", - "Training epoch 29...\n", - "Loss: 0.4752\n" - ] - } - ], + "outputs": [], "source": [ "for epoch in range(30):\n", " print(f'Training epoch {epoch}...')\n", @@ -224,7 +148,7 @@ }, { "cell_type": "code", - "execution_count": 20, + "execution_count": null, "id": "aa5b2497-90a7-44a9-9985-427b920945eb", "metadata": {}, "outputs": [], @@ -234,21 +158,10 @@ }, { "cell_type": "code", - "execution_count": 23, + "execution_count": null, "id": "ff62419f-ead5-4244-9a5e-a21d568bdca8", "metadata": {}, - "outputs": [ - { - "data": { - "text/plain": [ - "" - ] - }, - "execution_count": 23, - "metadata": {}, - "output_type": "execute_result" - } - ], + "outputs": [], "source": [ "net = NeuralNet()\n", "net.load_state_dict(torch.load('trained_net.pth'))" @@ -256,18 +169,10 @@ }, { "cell_type": "code", - "execution_count": 24, + "execution_count": null, "id": "eb1d4385-d4da-454f-9d8c-d844aa7b24df", "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Accuracy: 68.23%\n" - ] - } - ], + "outputs": [], "source": [ "correct = 0\n", "total = 0\n", @@ -286,20 +191,10 @@ }, { "cell_type": "code", - "execution_count": 32, + "execution_count": null, "id": "1a4f098f-b1ea-46b7-a7d6-769f80292f5f", "metadata": {}, - "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Prediction: cat\n", - "Prediction: deer\n", - "Prediction: plane\n" - ] - } - ], + "outputs": [], "source": [ "new_transform = transforms.Compose([\n", " transforms.Resize((32, 32)), \n", @@ -323,6 +218,14 @@ " _, predicted = torch.max(output, 1)\n", " print(f'Prediction: {class_names[predicted.item()]}')" ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "eed01777-2e7a-476c-8861-388d68f2f9c6", + "metadata": {}, + "outputs": [], + "source": [] } ], "metadata": {