File tree Expand file tree Collapse file tree 2 files changed +25
-0
lines changed
Expand file tree Collapse file tree 2 files changed +25
-0
lines changed Original file line number Diff line number Diff line change @@ -22,6 +22,10 @@ def prepare_model(
2222 data_dir = "./data" ,
2323 public_scope = Path (__file__ ).resolve ().parent / "public_scope.json" ,
2424):
25+ # TODO refactor this test so that it does not use eval
26+ # flake8: noqa: F401
27+ from model_api .models import ClassificationModel , DetectionModel , SegmentationModel
28+
2529 with open (public_scope , "r" ) as f :
2630 public_scope = json .load (f )
2731
Original file line number Diff line number Diff line change 66import numpy as np
77import onnx
88import pytest
9+
910from model_api .adapters .onnx_adapter import ONNXRuntimeAdapter
1011from model_api .adapters .openvino_adapter import OpenvinoAdapter , create_core
1112from model_api .adapters .utils import load_parameters_from_onnx
13+
14+ # TODO refactor this test so that it does not use eval
15+ # flake8: noqa: F401
1216from model_api .models import (
1317 ActionClassificationModel ,
18+ AnomalyDetection ,
1419 AnomalyResult ,
20+ ClassificationModel ,
1521 ClassificationResult ,
1622 DetectedKeypoints ,
23+ DetectionModel ,
1724 DetectionResult ,
25+ ImageModel ,
1826 ImageResultWithSoftPrediction ,
1927 InstanceSegmentationResult ,
28+ KeypointDetectionModel ,
29+ MaskRCNNModel ,
30+ PredictedMask ,
2031 Prompt ,
32+ SAMDecoder ,
33+ SAMImageEncoder ,
34+ SAMLearnableVisualPrompter ,
35+ SAMVisualPrompter ,
36+ SegmentationModel ,
2137 VisualPromptingResult ,
2238 ZSLVisualPromptingResult ,
2339 add_rotated_rects ,
2440 get_contours ,
2541)
42+ from model_api .tilers import (
43+ DetectionTiler ,
44+ InstanceSegmentationTiler ,
45+ SemanticSegmentationTiler ,
46+ )
2647
2748
2849def read_config (path : Path ):
You can’t perform that action at this time.
0 commit comments