|
|
import json |
|
|
import numpy as np |
|
|
|
|
|
def load_ids(filename): |
|
|
ids = open(filename).read().splitlines() |
|
|
ids = np.array(ids, dtype=np.int64) |
|
|
return ids |
|
|
|
|
|
|
|
|
|
|
|
class Instance(object): |
|
|
instance_id = 0 |
|
|
label_id = 0 |
|
|
vert_count = 0 |
|
|
med_dist = -1 |
|
|
dist_conf = 0.0 |
|
|
|
|
|
def __init__(self, mesh_vert_instances, instance_id): |
|
|
if (instance_id == -1): |
|
|
return |
|
|
self.instance_id = int(instance_id) |
|
|
self.label_id = int(self.get_label_id(instance_id)) |
|
|
self.vert_count = int(self.get_instance_verts(mesh_vert_instances, instance_id)) |
|
|
|
|
|
def get_label_id(self, instance_id): |
|
|
return int(instance_id // 1000) |
|
|
|
|
|
def get_instance_verts(self, mesh_vert_instances, instance_id): |
|
|
return (mesh_vert_instances == instance_id).sum() |
|
|
|
|
|
def to_json(self): |
|
|
return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4) |
|
|
|
|
|
def to_dict(self): |
|
|
dict = {} |
|
|
dict["instance_id"] = self.instance_id |
|
|
dict["label_id"] = self.label_id |
|
|
dict["vert_count"] = self.vert_count |
|
|
dict["med_dist"] = self.med_dist |
|
|
dict["dist_conf"] = self.dist_conf |
|
|
return dict |
|
|
|
|
|
def from_json(self, data): |
|
|
self.instance_id = int(data["instance_id"]) |
|
|
self.label_id = int(data["label_id"]) |
|
|
self.vert_count = int(data["vert_count"]) |
|
|
if ("med_dist" in data): |
|
|
self.med_dist = float(data["med_dist"]) |
|
|
self.dist_conf = float(data["dist_conf"]) |
|
|
|
|
|
def __str__(self): |
|
|
return "("+str(self.instance_id)+")" |
|
|
|
|
|
def get_instances(ids, class_ids, class_labels, id2label): |
|
|
instances = {} |
|
|
for label in class_labels: |
|
|
instances[label] = [] |
|
|
instance_ids = np.unique(ids) |
|
|
for id in instance_ids: |
|
|
if id == 0: |
|
|
continue |
|
|
inst = Instance(ids, id) |
|
|
if inst.label_id in class_ids: |
|
|
instances[id2label[inst.label_id]].append(inst.to_dict()) |
|
|
return instances |
|
|
|
