Update model link from ONNX model zoo (#1459)

yuwenzho · web-flow · commit 4ca8d1f940d7 · 2023-12-13T16:22:49.000+08:00
Signed-off-by: yuwenzho &lt;yuwen.zhou@intel.com&gt;
diff --git a/examples/onnxrt/body_analysis/onnx_model_zoo/arcface/quantization/ptq_static/prepare_model.py b/examples/onnxrt/body_analysis/onnx_model_zoo/arcface/quantization/ptq_static/prepare_model.py
@@ -6,7 +6,7 @@
 import onnx
 from onnx import version_converter
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/body_analysis/arcface/model/arcfaceresnet100-8.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/body_analysis/arcface/model/arcfaceresnet100-8.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/body_analysis/onnx_model_zoo/emotion_ferplus/quantization/ptq_static/prepare_model.py b/examples/onnxrt/body_analysis/onnx_model_zoo/emotion_ferplus/quantization/ptq_static/prepare_model.py
@@ -6,7 +6,7 @@
 import onnx
 from onnx import version_converter
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/body_analysis/emotion_ferplus/model/emotion-ferplus-8.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/body_analysis/emotion_ferplus/model/emotion-ferplus-8.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/body_analysis/onnx_model_zoo/ultraface/quantization/ptq_static/prepare_model.py b/examples/onnxrt/body_analysis/onnx_model_zoo/ultraface/quantization/ptq_static/prepare_model.py
@@ -6,7 +6,7 @@
 import onnx
 from onnx import version_converter
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/body_analysis/ultraface/models/version-RFB-320.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/body_analysis/ultraface/models/version-RFB-320.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/alexnet/model/bvlcalexnet-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/alexnet/model/bvlcalexnet-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/caffenet/model/caffenet-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/caffenet/model/caffenet-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/densenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/densenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/densenet-121/model/densenet-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/densenet-121/model/densenet-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/efficientnet-lite4/model/efficientnet-lite4-11.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/efficientnet-lite4/model/efficientnet-lite4-11.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/fcn/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/fcn/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/fcn/model/fcn-resnet50-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/fcn/model/fcn-resnet50-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/inception_and_googlenet/googlenet/model/googlenet-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/inception_and_googlenet/googlenet/model/googlenet-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/inception_and_googlenet/inception_v1/model/inception-v1-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/inception_and_googlenet/inception_v1/model/inception-v1-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/mnist/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/mnist/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/mnist/model/mnist-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/mnist/model/mnist-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/mobilenet/model/mobilenetv2-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/mobilenet/model/mobilenetv2-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/resnet/model/resnet50-v1-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/resnet/model/resnet50-v1-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/shufflenet/model/shufflenet-v2-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/shufflenet/model/shufflenet-v2-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/squeezenet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/squeezenet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/squeezenet/model/squeezenet1.0-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/squeezenet/model/squeezenet1.0-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/vgg/model/vgg16-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/vgg/model/vgg16-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq_static/prepare_model.py b/examples/onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/classification/zfnet-512/model/zfnet512-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/classification/zfnet-512/model/zfnet512-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlm/quantization/ptq_dynamic/main.py b/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlm/quantization/ptq_dynamic/main.py
@@ -29,6 +29,7 @@
 import onnx
 from neural_compressor.data import DataLoader
 
+import pyarrow_hotfix; pyarrow_hotfix.uninstall()
 
 # Will error if the minimal version of Transformers is not installed. Remove at your own risks.
 check_min_version("4.5.0")
diff --git a/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlm/quantization/ptq_static/main.py b/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlm/quantization/ptq_static/main.py
@@ -29,6 +29,7 @@
 import onnx
 from neural_compressor.data import DataLoader
 
+import pyarrow_hotfix; pyarrow_hotfix.uninstall()
 
 # Will error if the minimal version of Transformers is not installed. Remove at your own risks.
 check_min_version("4.5.0")
diff --git a/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlmv3/quantization/ptq_dynamic/main.py b/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlmv3/quantization/ptq_dynamic/main.py
@@ -37,6 +37,8 @@
 
 from neural_compressor.data import DataLoader
 
+import pyarrow_hotfix; pyarrow_hotfix.uninstall()
+
 logger = logging.getLogger(__name__)
 
 @dataclass
diff --git a/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlmv3/quantization/ptq_static/main.py b/examples/onnxrt/nlp/huggingface_model/token_classification/layoutlmv3/quantization/ptq_static/main.py
@@ -37,6 +37,8 @@
 
 from neural_compressor.data import DataLoader
 
+import pyarrow_hotfix; pyarrow_hotfix.uninstall()
+
 logger = logging.getLogger(__name__)
 
 @dataclass
diff --git a/examples/onnxrt/nlp/onnx_model_zoo/BiDAF/quantization/ptq_dynamic/prepare_model.py b/examples/onnxrt/nlp/onnx_model_zoo/BiDAF/quantization/ptq_dynamic/prepare_model.py
@@ -6,7 +6,7 @@
 import onnx
 from onnx import version_converter
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/text/machine_comprehension/bidirectional_attention_flow/model/bidaf-9.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/text/machine_comprehension/bidirectional_attention_flow/model/bidaf-9.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/nlp/onnx_model_zoo/bert-squad/quantization/ptq_dynamic/prepare_model.py b/examples/onnxrt/nlp/onnx_model_zoo/bert-squad/quantization/ptq_dynamic/prepare_model.py
@@ -8,7 +8,7 @@
     "bert":
     "https://storage.googleapis.com/bert_models/2018_10_18/uncased_L-12_H-768_A-12.zip",
     "BERT-Squad":
-    "https://github.com/onnx/models/raw/main/text/machine_comprehension/bert-squad/model/bertsquad-12.onnx",
+    "https://github.com/onnx/models/raw/main/archive/text/machine_comprehension/bert-squad/model/bertsquad-12.onnx",
 }
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/DUC/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/DUC/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/duc/model/ResNet101-DUC-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/duc/model/ResNet101-DUC-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/faster_rcnn/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/faster_rcnn/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/faster-rcnn/model/FasterRCNN-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/faster-rcnn/model/FasterRCNN-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/mask_rcnn/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/mask_rcnn/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/mask-rcnn/model/MaskRCNN-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/mask-rcnn/model/MaskRCNN-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/ssd/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/ssd/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/ssd/model/ssd-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/ssd/model/ssd-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/ssd_mobilenet_v1/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/ssd_mobilenet_v1/quantization/ptq_static/prepare_model.py
@@ -6,7 +6,7 @@
 import onnx
 from onnx import version_converter
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/ssd-mobilenetv1/model/ssd_mobilenet_v1_12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/ssd-mobilenetv1/model/ssd_mobilenet_v1_12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/tiny_yolov3/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/tiny_yolov3/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/tiny-yolov3/model/tiny-yolov3-11.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/tiny-yolov3/model/tiny-yolov3-11.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/yolov3/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/yolov3/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/yolov3/model/yolov3-12.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/yolov3/model/yolov3-12.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
diff --git a/examples/onnxrt/object_detection/onnx_model_zoo/yolov4/quantization/ptq_static/prepare_model.py b/examples/onnxrt/object_detection/onnx_model_zoo/yolov4/quantization/ptq_static/prepare_model.py
@@ -3,7 +3,7 @@
 import sys
 from urllib import request
 
-MODEL_URL = "https://github.com/onnx/models/raw/main/vision/object_detection_segmentation/yolov4/model/yolov4.onnx"
+MODEL_URL = "https://github.com/onnx/models/raw/main/archive/vision/object_detection_segmentation/yolov4/model/yolov4.onnx"
 MAX_TIMES_RETRY_DOWNLOAD = 5
 
 
