imlab.core.load_torch

imlab.core.load_torch(weight: str, classes: list, image_size: int) → Extractor[source]

load_torch. load pytorch model

Parameters

  • weight (str) –

  • classes (list) –

  • image_size (int) –

Return type

Extractor