Commit
•
06bf8ef
1
Parent(s):
62d77a1
Add tf.py verification printout (#5484)
Browse files* Update tf.py with verified confirmation
* Update ci-testing.yml
* Update ci-testing.yml
- models/tf.py +2 -0
models/tf.py
CHANGED
@@ -441,6 +441,8 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path
|
|
441 |
keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im))
|
442 |
keras_model.summary()
|
443 |
|
|
|
|
|
444 |
|
445 |
def parse_opt():
|
446 |
parser = argparse.ArgumentParser()
|
|
|
441 |
keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im))
|
442 |
keras_model.summary()
|
443 |
|
444 |
+
LOGGER.info('PyTorch, TensorFlow and Keras models successfully verified.\nUse export.py for TF model export.')
|
445 |
+
|
446 |
|
447 |
def parse_opt():
|
448 |
parser = argparse.ArgumentParser()
|