Unverified 提交 06bf8ef7 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Add tf.py verification printout (#5484)

* Update tf.py with verified confirmation * Update ci-testing.yml * Update ci-testing.yml
上级 62d77a10
...@@ -441,6 +441,8 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path ...@@ -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 = keras.Model(inputs=im, outputs=tf_model.predict(im))
keras_model.summary() keras_model.summary()
LOGGER.info('PyTorch, TensorFlow and Keras models successfully verified.\nUse export.py for TF model export.')
def parse_opt(): def parse_opt():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论