Browse Source
Add tf.py verification printout (#5484)
* Update tf.py with verified confirmation
* Update ci-testing.yml
* Update ci-testing.yml
modifyDataloader
Glenn Jocher
GitHub
3 years ago
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
2 additions and
0 deletions
-
models/tf.py
|
|
@@ -441,6 +441,8 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path |
|
|
|
keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im)) |
|
|
|
keras_model.summary() |
|
|
|
|
|
|
|
LOGGER.info('PyTorch, TensorFlow and Keras models successfully verified.\nUse export.py for TF model export.') |
|
|
|
|
|
|
|
|
|
|
|
def parse_opt(): |
|
|
|
parser = argparse.ArgumentParser() |