Skip to content

deeplabcut.pose_estimation_pytorch.modelzoo.inference

Classes:

Name Description
NumpyEncoder

Special json encoder for numpy types.

NumpyEncoder

Bases: JSONEncoder

Special json encoder for numpy types.

Source code in deeplabcut/pose_estimation_pytorch/modelzoo/inference.py
class NumpyEncoder(json.JSONEncoder):
    """Special json encoder for numpy types."""

    def default(self, obj):
        if isinstance(obj, np.ndarray):
            return obj.tolist()  # Convert ndarray to list
        return json.JSONEncoder.default(self, obj)