14 lines
491 B
TypeScript
14 lines
491 B
TypeScript
import { DetectedObject } from "@tensorflow-models/coco-ssd"
|
|
import ObjectDetector from "../ObjectDetector"
|
|
|
|
const defaultPredictions = [
|
|
(prediction: DetectedObject) => prediction.score > 0.6,
|
|
(prediction: DetectedObject) => prediction.class === 'person',
|
|
]
|
|
|
|
function makeObjectDetector (filterPredicates?: Function[]): ObjectDetector {
|
|
if (!filterPredicates) filterPredicates = defaultPredictions
|
|
return new ObjectDetector({ filterPredicates })
|
|
}
|
|
|
|
export default makeObjectDetector |