marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed cnn architecture, added website image upload

Browse Source
This commit is contained in:
marvin
2026-04-29 14:02:56 +02:00
parent fd1c111739
commit 262dbe9f98
18 changed files with 264 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bird_cnn/main.py
+3 -3
View File
@@ -34,13 +34,13 @@ val_size = len(dataset) - train_size
train_dataset, val_dataset = random_split(dataset, [train_size, val_size])
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=32, shuffle=False)
train_loader = DataLoader(train_dataset, batch_size=26, shuffle=True)
val_loader = DataLoader(val_dataset, batch_size=26, 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=15, c_out=7, kernel_size=3, img_width=IMAGE_SIZE[0], img_height=IMAGE_SIZE[1])
model = Bird_CNN(c_in=3, c_hidden=15, c_out=7)
model.to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=1e-3, weight_decay=1e-4)
loss_module = nn.CrossEntropyLoss()