code cleaning

2024-06-09 09:14:40 +02:00 · 2024-06-09 09:14:40 +02:00 · 856816e108
commit 856816e108
parent 03724c3cf4
3 changed files with 5 additions and 0 deletions
--- a/.vscode/extensions.json
+++ b/.vscode/extensions.json
@ -0,0 +1,3 @@
+{
+  "recommendations": ["sbsnippets.pytorch-snippets"]
+}
--- a/src/tile.py
+++ b/src/tile.py
@ -108,6 +108,8 @@ class Tile(pygame.sprite.Sprite):
        _, predicted = torch.max(output.data, 1)

        result = classes[predicted.item()]
+        
+
        if result == "ziemniak":
            result = 'marchew'
        return result
--- a/src/veggies_recognition/veggies/marchew_118.jpg
+++ b/src/veggies_recognition/veggies/marchew_118.jpg