Spaces:
Sleeping
Sleeping
Commit
·
7dfe4bc
1
Parent(s):
4eafc74
Update app.py
Browse files
app.py
CHANGED
|
@@ -48,12 +48,12 @@ def predict(img):
|
|
| 48 |
outputs = torch.argmax(outputs,1)
|
| 49 |
|
| 50 |
mask = np.array(outputs.cpu())
|
| 51 |
-
mask[mask==
|
| 52 |
-
mask[mask==1]=150
|
| 53 |
-
mask[mask==2]=76
|
| 54 |
-
mask[mask==2]=74
|
| 55 |
-
mask[mask==3]=
|
| 56 |
-
mask[mask==3]=
|
| 57 |
|
| 58 |
mask=np.reshape(mask,(480,640))
|
| 59 |
|
|
|
|
| 48 |
outputs = torch.argmax(outputs,1)
|
| 49 |
|
| 50 |
mask = np.array(outputs.cpu())
|
| 51 |
+
mask[mask==4]=255 #grape
|
| 52 |
+
mask[mask==1]=150 #leaves
|
| 53 |
+
mask[mask==2]=76 #pole
|
| 54 |
+
mask[mask==2]=74 #pole
|
| 55 |
+
mask[mask==3]=25 #wood
|
| 56 |
+
mask[mask==3]=29 #wood
|
| 57 |
|
| 58 |
mask=np.reshape(mask,(480,640))
|
| 59 |
|