tensorflow2

2020-03-25 18:23:00 -03:00
parent 7010af8a58
commit 7cf0c577a1
25 changed files with 1016 additions and 309 deletions
--- a/keras-yolo3-master/utils/multi_gpu_model.py
+++ b/keras-yolo3-master/utils/multi_gpu_model.py
@@ -1,5 +1,5 @@
-from keras.layers import Lambda, concatenate
-from keras.models import Model
+from tensorflow.keras.layers import Lambda, concatenate
+from tensorflow.keras.models import Model
 import tensorflow as tf

 def multi_gpu_model(model, gpus):
@@ -59,4 +59,4 @@ def multi_gpu_model(model, gpus):
        for name, outputs in zip(model.output_names, all_outputs):
            merged.append(concatenate(outputs,
                                                                axis=0, name=name))
-        return Model(model.inputs, merged)
+        return Model(model.inputs, merged)