Commit ·
377fcf0
1
Parent(s): b78bb63
Finally fix mobilenet error
Browse files- src/nets/mobilenet.py +1 -1
src/nets/mobilenet.py
CHANGED
|
@@ -98,7 +98,7 @@ def MobileNet(input_shape=None,
|
|
| 98 |
return model
|
| 99 |
|
| 100 |
def relu6(x):
|
| 101 |
-
return ReLU(
|
| 102 |
|
| 103 |
if __name__ == '__main__':
|
| 104 |
model = MobileNet(input_shape=(224, 224, 3))
|
|
|
|
| 98 |
return model
|
| 99 |
|
| 100 |
def relu6(x):
|
| 101 |
+
return ReLU(max_value=6)(x)
|
| 102 |
|
| 103 |
if __name__ == '__main__':
|
| 104 |
model = MobileNet(input_shape=(224, 224, 3))
|