Method test_detection_model

tests/test_high_level_api.py:108–116

Test DetectionModel base class.

(self)

        assert object_annotation3.category.name == category_name

    def test_detection_model(self) -> None:
        """Test DetectionModel base class."""
        from sahi import DetectionModel

        MODEL_PATH = "model_path"
        IMAGE_SIZE = 640
        detection_model = DetectionModel(model_path="model_path", image_size=IMAGE_SIZE, load_at_init=False)
        assert detection_model.model_path == MODEL_PATH
        assert detection_model.image_size == IMAGE_SIZE