s483780/inteligenty-traktor
s483780/inteligenty-traktor
3
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

code cleaning

Browse Source
This commit is contained in:
Zofia Lorenc 2024-06-09 09:14:40 +02:00
parent 03724c3cf4
commit 856816e108
3 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.vscode/extensions.json vendored Normal file
View File

@ -0,0 +1,3 @@
{
"recommendations": ["sbsnippets.pytorch-snippets"]
}

2
src/tile.py
View File

@ -108,6 +108,8 @@ class Tile(pygame.sprite.Sprite):
_, predicted = torch.max(output.data, 1) _, predicted = torch.max(output.data, 1)
result = classes[predicted.item()] result = classes[predicted.item()]
if result == "ziemniak": if result == "ziemniak":
result = 'marchew' result = 'marchew'
return result return result

BIN
src/veggies_recognition/veggies/marchew_118.jpg
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 9.2 KiB