Tim77777767
commited on
Commit
·
8ac071c
1
Parent(s):
448673e
Anpassungen an preTrained
Browse files- preTrainedTest.py +2 -0
preTrainedTest.py
CHANGED
|
@@ -30,6 +30,8 @@ transform = T.Compose([
|
|
| 30 |
])
|
| 31 |
input_tensor = transform(image).unsqueeze(0).to(device)
|
| 32 |
|
|
|
|
|
|
|
| 33 |
# Inferenz
|
| 34 |
with torch.no_grad():
|
| 35 |
output = model(input_tensor)
|
|
|
|
| 30 |
])
|
| 31 |
input_tensor = transform(image).unsqueeze(0).to(device)
|
| 32 |
|
| 33 |
+
print("Modell geladen, Bild geladen, Preprocessing abgeschlossen")
|
| 34 |
+
|
| 35 |
# Inferenz
|
| 36 |
with torch.no_grad():
|
| 37 |
output = model(input_tensor)
|