Skip to content

Commit e809f4e

Browse files
fix imports
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent 3262502 commit e809f4e

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

tests/python/accuracy/test_accuracy.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,18 +15,35 @@
1515
# flake8: noqa: F401
1616
from model_api.models import (
1717
ActionClassificationModel,
18+
AnomalyDetection,
1819
AnomalyResult,
20+
ClassificationModel,
1921
ClassificationResult,
2022
DetectedKeypoints,
23+
DetectionModel,
2124
DetectionResult,
25+
ImageModel,
2226
ImageResultWithSoftPrediction,
2327
InstanceSegmentationResult,
28+
KeypointDetectionModel,
29+
MaskRCNNModel,
30+
PredictedMask,
2431
Prompt,
32+
SAMDecoder,
33+
SAMImageEncoder,
34+
SAMLearnableVisualPrompter,
35+
SAMVisualPrompter,
36+
SegmentationModel,
2537
VisualPromptingResult,
2638
ZSLVisualPromptingResult,
2739
add_rotated_rects,
2840
get_contours,
2941
)
42+
from model_api.tilers import (
43+
DetectionTiler,
44+
InstanceSegmentationTiler,
45+
SemanticSegmentationTiler,
46+
)
3047

3148

3249
def read_config(path: Path):

0 commit comments

Comments
 (0)