Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]fix onnx thread error

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

    * fix onnx thread error
This commit is contained in:
dangwei.ldw
2022-08-09 17:46:24 +08:00
committed by wenmeng.zwm
parent 1c10c24f82
commit 4314ecf66b
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
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

4
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(