Skip to main content
Version: 2.0.0

Base Model

개요

  • NuFiStreamer의 모델에 대한 인터페이스를 정의하는 추상 클래스입니다.
  • NuFiStreamer에서 제공하는 모든 모델은 BaseModel을 상속합니다.

Interface

  • process(self, image: numpy.ndarray, pipeline_id: int = None)
    • 이미지 프레임을 처리하는 함수
    • Input
      • image: 모델 처리 대상 이미지
      • pipeline_id: 해당 모델을 이용하는 pipeline의 id (optional). 모델을 여러 파이프라인에서 공유할경우 각 파이프라인을 구분하기위해 사용됩니다.
    • Output
  • clean(self, pipeline_id: int)
    • nufi element에서 파이프라인의 state가 NULL로 바뀔 때 실행되는 함수
    • Input
      • pipeline_id: 해당 모델을 이용하는 element가 속한 pipeline id
    • Output
      • None