Download preconverted yolo-v2-tiny and yolo-v4-tiny-tf (#3894)

Ticket 125321

Author: Wovchena

models/public/yolo-v2-tiny-tf/model.yml

@@ -19,55 +19,10 @@ description: >-
   (COCO) <https://cocodataset.org/#home> dataset with 80 classes.
 task_type: detection
 files:
-  - name: yolov2-tiny.weights
-    size: 44948600
-    checksum: 48109100a3414fce39d57c8dd1feaee8714423e473f9d4dfd2cf8e1aa23f7e0e5920b9362ad7e3c804f24522e06fe536
-    original_source: https://pjreddie.com/media/files/yolov2-tiny.weights
-    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/yolo-v2-tiny-tf/yolov2-tiny.weights
-  - name: keras-YOLOv3-model-set/tools/model_converter/convert.py
-    size: 12937
-    checksum: fd6d1550e3843bc4ae27d1cf53ade56606a01c2162d149bc58ed8d654dfbbe59754a2bfd8e8ca508ecee663b38f6e7bf
-    original_source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/tools/model_converter/convert.py
-    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/yolo-v2-tiny-tf/convert.py
-  - name: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    size: 8370
-    checksum: 1be7f22eafa2c9fd54457938bebbe4d572b7b2abdfed9a1fd720bbb52cdc96151839726957aa8c7dde9cd3abb332807e
-    original_source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/tools/model_converter/keras_to_tensorflow.py
-    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/yolo-v2-tiny-tf/keras_to_tensorflow.py
-  - name: keras-YOLOv3-model-set/common/utils.py
-    size: 5002
-    checksum: 7acd0f55fb4c1a0f00df46c2383309bc12715d66c09adb9bcb22d6aea2555ef0478d2d7147fe5cd8de770a23008e4958
-    original_source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/common/utils.py
-    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/yolo-v2-tiny-tf/utils.py
-  - name: keras-YOLOv3-model-set/cfg/yolov2-tiny.cfg
-    size: 1488
-    checksum: bb52511d6b3c33a066bcd2e41a86d300c854e73b74b9064bb5bfe510ef8801315228abc5e5a89e7dacb3e0c330ed07cd
-    original_source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/cfg/yolov2-tiny.cfg
-    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/yolo-v2-tiny-tf/yolov2-tiny.cfg
-postprocessing:
-  # disable imports that aren't needed for this model and code that uses them
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/tools/model_converter/convert.py
-    pattern: 'from yolo4\.'
-    replacement: '# \g<0>'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/common/utils.py
-    pattern: 'import cv2,|from (matplotlib|PIL|common\.backbones|yolo4)\b'
-    replacement: '# \g<0>'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/common/utils.py
-    pattern: '''(swish|hard_\w+|mish)'''
-    replacement: '# \g<0>'
-
-  # patch TensorFlow 2 compatibility
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    pattern: 'import tensorflow as tf'
-    replacement: 'import tensorflow.compat.v1 as tf\ntf.disable_v2_behavior()'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    pattern: 'from tensorflow\.keras\b'
-    replacement: 'from tensorflow.compat.v1.keras'
+  - name: yolo-v2-tiny.pb
+    size: 44963146
+    checksum: 50636a68eec6799007e4a013c8d89ae5fc16803cb3872465e16daaef758be0309a179824fda189d544cd0da313457f25
+    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2024.0/yolo-v2-tiny-tf/yolo-v2-tiny.pb
 input_info:
   - name: image_input
     shape: [1, 416, 416, 3]

models/public/yolo-v2-tiny-tf/pre-convert.py

@@ -21,54 +21,10 @@ description: >-
   dataset with 80 classes.
 task_type: detection
 files:
-  - name: yolov4-tiny.weights
-    size: 24251276
-    checksum: ee417b9217aa806b93110e8c688c06f7d75778645372667712f989fc961175244162d9a74f402dfedc64b80b3a21e58d
-    source: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights
-  - name: keras-YOLOv3-model-set/tools/model_converter/convert.py
-    size: 16060
-    checksum: 6a72c81fce395d3d7f16b713fa70e539aabe53403baac7f6f1e7a51ad11e50cc37a708b96d9b239c89d1a1210d8edd7d
-    source: https://github.com/david8862/keras-YOLOv3-model-set/raw/6b4a0ee63771262363e8224b0ee915cad6c5e93e/tools/model_converter/convert.py
-  - name: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    size: 8370
-    checksum: 1be7f22eafa2c9fd54457938bebbe4d572b7b2abdfed9a1fd720bbb52cdc96151839726957aa8c7dde9cd3abb332807e
-    source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/tools/model_converter/keras_to_tensorflow.py
-  - name: keras-YOLOv3-model-set/common/utils.py
-    size: 5002
-    checksum: 7acd0f55fb4c1a0f00df46c2383309bc12715d66c09adb9bcb22d6aea2555ef0478d2d7147fe5cd8de770a23008e4958
-    source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/common/utils.py
-  - name: keras-YOLOv3-model-set/yolo4/models/layers.py
-    size: 15171
-    checksum: b949a66aeac84fe0ff6990961c8b39d1dc6585e4ce206c2fbafa5ee5f1063f7a55424beeacf55932bf8afe621c308a3e
-    source: https://github.com/david8862/keras-YOLOv3-model-set/raw/d38c3d865f7190ee9b19a30e91f2b750a31320c1/yolo4/models/layers.py
-  - name: keras-YOLOv3-model-set/cfg/yolov4-tiny.cfg
-    size: 3028
-    checksum: 387ffb240df5f9c81355145a72a5d6bba085db3118b0f61ad214d3d3491e957f3b3ab9498eb8b873c098404fa5235cb7
-    source: https://github.com/david8862/keras-YOLOv3-model-set/raw/6b4a0ee63771262363e8224b0ee915cad6c5e93e/cfg/yolov4-tiny.cfg
-postprocessing:
-  # disable imports that aren't needed for this model and code that uses them
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/common/utils.py
-    pattern: 'import cv2,|from (matplotlib|PIL|common\.backbones)\b'
-    replacement: '# \g<0>'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/common/utils.py
-    pattern: '''(swish|hard_\w+)'''
-    replacement: '# \g<0>'
-
-  # patch TensorFlow 2 compatibility
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    pattern: 'import tensorflow as tf'
-    replacement: 'import tensorflow.compat.v1 as tf\ntf.disable_v2_behavior()'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/tools/model_converter/keras_to_tensorflow.py
-    pattern: 'from tensorflow\.keras\b'
-    replacement: 'from tensorflow.compat.v1.keras'
-  - $type: regex_replace
-    file: keras-YOLOv3-model-set/yolo4/models/layers.py
-    pattern: 'tensorflow\.keras'
-    replacement: 'tensorflow.compat.v1.keras'
+  - name: yolo-v4-tiny.pb
+    size: 24287302
+    checksum: bc9ccc0a3d8f137afe377b1e845d17fde00c19a103c9ee077519ce25d3b7f4e68fcea3940f09737d11fa4ca02a035bd7
+    source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2024.0/yolo-v4-tiny-tf/yolo-v4-tiny.pb
 input_info:
   - name: image_input
     shape: [1, 416, 416, 3]