diff --git a/.vscode/extensions.json b/.vscode/extensions.json new file mode 100644 index 00000000..5ccd0950 --- /dev/null +++ b/.vscode/extensions.json @@ -0,0 +1,3 @@ +{ + "recommendations": ["sbsnippets.pytorch-snippets"] +} diff --git a/src/tile.py b/src/tile.py index 71c82cee..41ab871c 100644 --- 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 diff --git a/src/veggies_recognition/veggies/marchew_118.jpg b/src/veggies_recognition/veggies/marchew_118.jpg deleted file mode 100644 index 62ba9ca7..00000000 Binary files a/src/veggies_recognition/veggies/marchew_118.jpg and /dev/null differ