File size: 651 Bytes
5626a1a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from typing import List, Dict, Optional
from .ImageMetadata import ImageMetadata
class AnnotatedData:
def __init__(self, data: Dict):
self.image_ids = data.get("_via_image_id_list", [])
self.metadata = {
img_id: ImageMetadata(
data["_via_img_metadata"][img_id]["filename"],
data["_via_img_metadata"][img_id]["size"],
data["_via_img_metadata"][img_id].get("regions", [])
) for img_id in self.image_ids if img_id in data["_via_img_metadata"]
}
def __repr__(self):
return f"AnnotatedData(Images={list(self.metadata.keys())})" |