콘텐츠로 이동

Visual Anomaly Inference

cycle_counter.VisualAnomalyInferenceHandler

build(config) classmethod

InferenceHandler 인스턴스를 생성합니다.

Parameters:

Name Type Description Default
config dict

InferenceHandler를 build 하기 위한 config가 담긴 dictionary 입니다.

{
    "models": list[dict],  # 모델 정보
        ```python
        {
            "checkpoint_path": str,  # 체크포인트 경로
            "password": str,  # 체크포인트 패스워드
            "device": str,  # 디바이스
            "model_options": dict,  # 모델 옵션
            "model_type": str,  # 모델 타입 (det, iad)
            "model_id": int,  # 모델 아이디 (Unique, 중복 불가)
            "inference_target": dict,  # 인퍼런스 타겟
                ```python
                {
                    "preceding_model": str,  # 이전 모델, (window_side_arm, window_side_index, aisle_side_arm, aisle_side_index)
                    "targets": list[dict],  # 인퍼런스 타겟
                        ```python
                        [
                            {"class_id": int, "offset": int},
                        ]
                        ```
                }
                ```
        }
        ```
}

required

Returns:

Name Type Description
InferenceHandler VisualAnomalyInferenceHandler

build가 완료된 InferenceHandler 인스턴스를 반환합니다.

VisualAnomalyInferenceHandler

```

infer(data)

InferenceHandler를 사용하여 데이터를 추론합니다.

set_inference_option(model_id, key, value)

InferenceHandler를 사용하여 추론 옵션을 설정합니다.

Parameters:

Name Type Description Default
model_id int

int, 모델 아이디

required
key str

str, 옵션 key

required
value Any

Any, 옵션 value

required

Returns:

Type Description
None

None

get_inference_option(model_id, key)

InferenceHandler를 사용하여 추론 옵션을 가져옵니다.

Parameters:

Name Type Description Default
model_id int

int, 모델 아이디

required
key str

str, 옵션 key

required

Returns:

Type Description
Any

get_default_inference_option과 동일합니다.

Keys

각 모델의 get_default_inference_option과 동일합니다.

get_default_inference_option(model_id, key)

InferenceHandler를 사용하여 기본 추론 옵션을 가져옵니다.

Parameters:

Name Type Description Default
model_id int

int, 모델 아이디

required
key str

str, 옵션 key

required

Returns:

Type Description
Any

get_default_inference_option과 동일합니다.

Keys

각 모델의 get_default_inference_option과 동일합니다.

warmup()

InferenceHandler를 사용하여 추론 옵션을 웜업합니다.

Returns:

Type Description
None

None