Skip to content

Commit bd23487

Browse files
committed
Add yolo specific coco labels
1 parent c3b5270 commit bd23487

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

packages/react-native-executorch/src/modules/computer_vision/InstanceSegmentationModule.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,13 @@ import {
88
SegmentedInstance,
99
InstanceSegmentationOptions,
1010
} from '../../types/instanceSegmentation';
11-
import { CocoLabel } from '../../types/objectDetection';
1211
import { RnExecutorchErrorCode } from '../../errors/ErrorCodes';
1312
import { RnExecutorchError } from '../../errors/errorUtils';
1413
import { BaseLabeledModule, fetchModelPath } from '../BaseLabeledModule';
14+
import { CocoLabelYolo } from '../../constants/commonVision';
1515

16-
const YOLO_SEG_CONFIG: InstanceSegmentationConfig<typeof CocoLabel> = {
17-
labelMap: CocoLabel,
16+
const YOLO_SEG_CONFIG: InstanceSegmentationConfig<typeof CocoLabelYolo> = {
17+
labelMap: CocoLabelYolo,
1818
availableInputSizes: [384, 512, 640] as const,
1919
defaultInputSize: 384,
2020
defaultConfidenceThreshold: 0.5,

0 commit comments

Comments
 (0)