Browse Source

[to #42322933]fix onnx thread error

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9669460

    * fix onnx thread error
master
dangwei.ldw yingda.chen 3 years ago
parent
commit
5fcb8e2342
2 changed files with 7 additions and 3 deletions
  1. +5
    -1
      modelscope/models/cv/product_retrieval_embedding/item_detection.py
  2. +2
    -2
      tests/pipelines/test_product_retrieval_embedding.py

+ 5
- 1
modelscope/models/cv/product_retrieval_embedding/item_detection.py View File

@@ -21,7 +21,11 @@ class YOLOXONNX(object):
self.num_classes = 13
self.onnx_path = onnx_path
import onnxruntime as ort
self.ort_session = ort.InferenceSession(self.onnx_path)
options = ort.SessionOptions()
options.intra_op_num_threads = 1
options.inter_op_num_threads = 1
self.ort_session = ort.InferenceSession(
self.onnx_path, sess_options=options)
self.with_p6 = False
self.multi_detect = multi_detect



+ 2
- 2
tests/pipelines/test_product_retrieval_embedding.py View File

@@ -13,14 +13,14 @@ class ProductRetrievalEmbeddingTest(unittest.TestCase):
model_id = 'damo/cv_resnet50_product-bag-embedding-models'
img_input = 'data/test/images/product_embed_bag.jpg'

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_model_name(self):
product_embed = pipeline(Tasks.product_retrieval_embedding,
self.model_id)
result = product_embed(self.img_input)[OutputKeys.IMG_EMBEDDING]
print('abs sum value is: {}'.format(np.sum(np.abs(result))))

@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
def test_run_with_model_from_modelhub(self):
model = Model.from_pretrained(self.model_id)
product_embed = pipeline(


Loading…
Cancel
Save