Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def predict_2(image):
|
|
28 |
|
29 |
image = Image.fromarray(image).resize((size,size))
|
30 |
image = np.array(image)
|
31 |
-
stride =
|
32 |
steps = int(pach_size/stride)
|
33 |
patches_img = patchify(image, (pach_size, pach_size, 3), step=steps) #Step=256 for 256 patches means no overlap
|
34 |
patches_img = patches_img[:,:,0,:,:,:]
|
@@ -89,16 +89,16 @@ targets_classes_colors = np.array([[ 0, 0, 0],
|
|
89 |
[ 2, 135, 115],
|
90 |
[255, 0, 0]])
|
91 |
|
92 |
-
class_weights = {0: 1
|
93 |
-
1: 1
|
94 |
2: 2.171655596616696,
|
95 |
-
3: 1
|
96 |
-
4: 1
|
97 |
5: 2.2101197049812593,
|
98 |
6: 11.601519937899578,
|
99 |
7: 7.99072122367673,
|
100 |
-
8: 1
|
101 |
-
9: 1
|
102 |
10: 2.5426918173402457,
|
103 |
11: 11.187574445057574,
|
104 |
12: 241.57620214903147,
|
|
|
28 |
|
29 |
image = Image.fromarray(image).resize((size,size))
|
30 |
image = np.array(image)
|
31 |
+
stride = 1
|
32 |
steps = int(pach_size/stride)
|
33 |
patches_img = patchify(image, (pach_size, pach_size, 3), step=steps) #Step=256 for 256 patches means no overlap
|
34 |
patches_img = patches_img[:,:,0,:,:,:]
|
|
|
89 |
[ 2, 135, 115],
|
90 |
[255, 0, 0]])
|
91 |
|
92 |
+
class_weights = {0: 0.1,
|
93 |
+
1: 0.1,
|
94 |
2: 2.171655596616696,
|
95 |
+
3: 0.1,
|
96 |
+
4: 0.1,
|
97 |
5: 2.2101197049812593,
|
98 |
6: 11.601519937899578,
|
99 |
7: 7.99072122367673,
|
100 |
+
8: 0.1,
|
101 |
+
9: 0.1,
|
102 |
10: 2.5426918173402457,
|
103 |
11: 11.187574445057574,
|
104 |
12: 241.57620214903147,
|