Commit a2d2d1f 1 parent 08c50e1 commit a2d2d1f Copy full SHA for a2d2d1f
File tree 4 files changed +17
-8
lines changed
4 files changed +17
-8
lines changed Original file line number Diff line number Diff line change @@ -2,3 +2,5 @@ export * from './ETModule';
2
2
export * from './RnExecutorch' ;
3
3
export * from './StyleTransfer' ;
4
4
export * from './constants/modelUrls' ;
5
+ export * from './models/object_detection/ObjectDetection' ;
6
+ export * from './models/object_detection/types' ;
Original file line number Diff line number Diff line change @@ -24,13 +24,14 @@ export const useObjectDetection = ({
24
24
25
25
useEffect ( ( ) => {
26
26
const loadModel = async ( ) => {
27
+ let path = modelSource ;
27
28
if ( typeof modelSource === 'number' ) {
28
- modelSource = Image . resolveAssetSource ( modelSource ) . uri ;
29
+ path = Image . resolveAssetSource ( modelSource ) . uri ;
29
30
}
30
31
31
32
try {
32
33
setIsModelLoading ( true ) ;
33
- await ObjectDetection . loadModule ( modelSource ) ;
34
+ await ObjectDetection . loadModule ( path ) ;
34
35
} catch ( e ) {
35
36
setError ( getError ( e ) ) ;
36
37
} finally {
Original file line number Diff line number Diff line change 1
- export interface Bbox {
1
+ export interface Detection {
2
2
x1 : number ;
3
- y1 : number ;
4
3
x2 : number ;
4
+ y1 : number ;
5
5
y2 : number ;
6
- }
7
-
8
- export interface Detection {
9
- bbox : Bbox ;
10
6
label : keyof typeof CocoLabel ;
11
7
score : number ;
12
8
}
Original file line number Diff line number Diff line change
1
+ import type { TurboModule } from 'react-native' ;
2
+ import { TurboModuleRegistry } from 'react-native' ;
3
+ import { ObjectDetectionResult } from '../models/object_detection/types' ;
4
+
5
+ export interface Spec extends TurboModule {
6
+ loadModule ( modelSource : string ) : Promise < number > ;
7
+ forward ( input : string ) : Promise < ObjectDetectionResult > ;
8
+ }
9
+
10
+ export default TurboModuleRegistry . get < Spec > ( 'ObjectDetection' ) ;
You can’t perform that action at this time.
0 commit comments