Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import numpy as np
|
2 |
|
3 |
-
# Patch np.sctypes for imgaug to work
|
4 |
if not hasattr(np, 'sctypes'):
|
5 |
np.sctypes = {
|
6 |
-
"float": [np.float16, np.float32, np.float64]
|
|
|
7 |
}
|
8 |
|
9 |
# Now, import imgaug after patching
|
|
|
1 |
import numpy as np
|
2 |
|
|
|
3 |
if not hasattr(np, 'sctypes'):
|
4 |
np.sctypes = {
|
5 |
+
"float": [np.float16, np.float32, np.float64],
|
6 |
+
"int": [np.int8, np.int16, np.int32, np.int64]
|
7 |
}
|
8 |
|
9 |
# Now, import imgaug after patching
|