test
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def predict(inp):
|
|
| 53 |
cam = GradCAM(model=model,
|
| 54 |
target_layers=target_layers)
|
| 55 |
img = ho_trans_center(image = inp)['image']
|
| 56 |
-
rgb_img = img.copy()
|
| 57 |
img = topt(image = img)['image']
|
| 58 |
img = img.unsqueeze(0)
|
| 59 |
with torch.no_grad():
|
|
|
|
| 53 |
cam = GradCAM(model=model,
|
| 54 |
target_layers=target_layers)
|
| 55 |
img = ho_trans_center(image = inp)['image']
|
| 56 |
+
rgb_img = img.copy()/255.
|
| 57 |
img = topt(image = img)['image']
|
| 58 |
img = img.unsqueeze(0)
|
| 59 |
with torch.no_grad():
|