serialize paths

Author: Matt Sokoloff

labelbox/data/serialization/coco/annotation.py

@@ -5,6 +5,8 @@
 from pydantic import BaseModel
 import numpy as np
 
+from .path import PathSerializerMixin
+
 
 def rle_decoding(rle_arr: List[int], w: int, h: int) -> np.ndarray:
     indices = []
@@ -50,7 +52,7 @@ class COCOObjectAnnotation(BaseModel):
     iscrowd: int = 0
 
 
-class PanopticAnnotation(BaseModel):
+class PanopticAnnotation(PathSerializerMixin):
     # One to one relationship between image and panoptic annotation
     image_id: int
     file_name: Path

labelbox/data/serialization/coco/converter.py

@@ -1,16 +1,11 @@
 from typing import Dict, Any, Union
-from pathlib import Path, PosixPath, WindowsPath
+from pathlib import Path
 import os
 
-import pydantic.json
-
 from labelbox.data.annotation_types.collection import LabelCollection, LabelGenerator
 from labelbox.data.serialization.coco.instance_dataset import CocoInstanceDataset
 from labelbox.data.serialization.coco.panoptic_dataset import CocoPanopticDataset
 
-pydantic.json.ENCODERS_BY_TYPE[PosixPath] = str
-pydantic.json.ENCODERS_BY_TYPE[WindowsPath] = str
-
 
 def create_path_if_not_exists(path: Union[Path, str],
                               ignore_existing_data=False):

labelbox/data/serialization/coco/image.py

@@ -1,14 +1,14 @@
 from pathlib import Path
 
 from typing import Optional, Tuple
-from pydantic import BaseModel
 from PIL import Image
 import imagesize
 
+from .path import PathSerializerMixin
 from labelbox.data.annotation_types import Label
 
 
-class CocoImage(BaseModel):
+class CocoImage(PathSerializerMixin):
     id: int
     width: int
     height: int