def set_pic_label_worker(con): if cluster.get_size() == 0: # TODO: Remove cluster like that??? embeddings_row_dicts = DBManager.remove_cluster(cluster, con=con, close_connections=False) emb_id_to_face_dict = make_emb_id_to_face_dict_from_row_dicts(embeddings_row_dicts) emb_id_to_img_id_dict = make_emb_id_to_img_id_dict_from_row_dicts(embeddings_row_dicts) else: emb_id_to_face_dict = None emb_id_to_img_id_dict = None DBManager.store_clusters(modified_clusters, emb_id_to_face_dict=emb_id_to_face_dict, emb_id_to_img_id_dict=emb_id_to_img_id_dict, con=con, close_connections=False) DBManager.store_certain_labels(cluster=new_cluster, con=con, close_connections=False)
def set_cluster_label_worker(con): DBManager.store_clusters([cluster], con=con, close_connections=False) cluster.set_label(new_label)