From 58af9f0f82ce933d367bcc39003c8b07a0d106c1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marvin=20Krau=C3=9Fer?= Date: Tue, 12 May 2026 04:32:19 +0200 Subject: [PATCH] changed dockerfile and bird dataset --- api/Dockerfile | 2 +- api/train_bird.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/api/Dockerfile b/api/Dockerfile index e1ae089..2a9b54a 100644 --- a/api/Dockerfile +++ b/api/Dockerfile @@ -12,7 +12,7 @@ COPY requirements.txt . RUN pip install --no-cache-dir -r requirements.txt -Copy . . +COPY . . RUN chmod +x start_server.sh diff --git a/api/train_bird.py b/api/train_bird.py index 9b3baab..4f4ad9a 100644 --- a/api/train_bird.py +++ b/api/train_bird.py @@ -24,12 +24,14 @@ class bird_species(Enum): Sarus_Crane = 6 transform = transforms.Compose([ + transforms.RandomHorizontalFlip(p=0.5), + transforms.RandomRotation(35), transforms.Resize(IMAGE_SIZE), transforms.CenterCrop(IMAGE_SIZE), transforms.ToTensor() ]) -dataset = datasets.ImageFolder("data/train", transform=transform) +dataset = datasets.ImageFolder("data/CUB_200_2011/images", transform=transform) train_size = int(0.8 * len(dataset)) val_size = len(dataset) - train_size @@ -43,7 +45,7 @@ val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False) device = torch.device("cpu") if not torch.cuda.is_available() else torch.device("cuda:0") print("Using device", device) -model = Bird_CNN(c_in=3, c_hidden=16, c_out=7) +model = Bird_CNN(c_in=3, c_hidden=16, c_out=200) model.to(device) optimizer = torch.optim.Adam(model.parameters(), lr=1e-3, weight_decay=1e-4) loss_module = nn.CrossEntropyLoss()