|
6 | 6 | from fastvisionops.bbox import nms as bbox_compat_nms |
7 | 7 | from fastvisionops.build import SOURCE, build_native_backend |
8 | 8 | from fastvisionops.mask import mask_nms as mask_compat_nms |
9 | | -from fastvisionops.native import NativeBackend |
10 | | -from fastvisionops.native.backend import NativeBackend as BackendImplementation |
| 9 | +from fastvisionops.native import ( |
| 10 | + DEFAULT_OUTPUT, |
| 11 | + NativeBackend, |
| 12 | + hwc_to_chw_normalize as native_hwc_to_chw_normalize, |
| 13 | + hwc_to_chw_normalize_batched as native_hwc_to_chw_normalize_batched, |
| 14 | +) |
| 15 | +from fastvisionops.native.backend import ( |
| 16 | + NativeBackend as BackendImplementation, |
| 17 | + hwc_to_chw_normalize as native_hwc_to_chw_normalize_implementation, |
| 18 | + hwc_to_chw_normalize_batched as native_batched_implementation, |
| 19 | +) |
11 | 20 | from fastvisionops.native.build import ( |
| 21 | + DEFAULT_OUTPUT as native_default_output, |
12 | 22 | build_native_backend as native_build_native_backend, |
13 | 23 | ) |
14 | 24 | from fastvisionops.postprocess import mask_nms as postprocess_mask_nms |
@@ -47,6 +57,15 @@ def test_mask_import_paths_share_one_implementation(self): |
47 | 57 | def test_preprocess_and_native_packages_expose_implementations(self): |
48 | 58 | self.assertIs(hwc_to_chw_normalize, numpy_hwc_to_chw_normalize) |
49 | 59 | self.assertIs(NativeBackend, BackendImplementation) |
| 60 | + self.assertIs( |
| 61 | + native_hwc_to_chw_normalize, |
| 62 | + native_hwc_to_chw_normalize_implementation, |
| 63 | + ) |
| 64 | + self.assertIs( |
| 65 | + native_hwc_to_chw_normalize_batched, |
| 66 | + native_batched_implementation, |
| 67 | + ) |
| 68 | + self.assertIs(DEFAULT_OUTPUT, native_default_output) |
50 | 69 | self.assertIs(build_native_backend, native_build_native_backend) |
51 | 70 |
|
52 | 71 | def test_native_source_is_colocated_with_backend(self): |
|
0 commit comments