Python add_cluster_neighbors Examples

Example #1

File: create_submodels.py Project: mapillary/OpenSfM

    def _add_cluster_neighbors(self, meta_data, max_distance):
        images, positions, labels, centers = meta_data.load_clusters()
        clusters = tools.add_cluster_neighbors(
            positions, labels, centers, max_distance)

        image_clusters = []
        for cluster in clusters:
            image_clusters.append(list(np.take(images, np.array(cluster))))

        meta_data.save_clusters_with_neighbors(image_clusters)

Example #2

File: create_submodels.py Project: AaronQi0731/OpenSfM

    def _add_cluster_neighbors(self, meta_data, max_distance):
        images, positions, labels, centers = meta_data.load_clusters()
        clusters = tools.add_cluster_neighbors(positions, labels, centers,
                                               max_distance)

        image_clusters = []
        for cluster in clusters:
            image_clusters.append(list(np.take(images, np.array(cluster))))

        meta_data.save_clusters_with_neighbors(image_clusters)

Example #3

File: submodel_creation.py Project: jchen706/grpc_stages

def _add_cluster_neighbors(file_path, max_distance):
    images, positions, labels, centers = load_clusters(file_path)
    clusters = tools.add_cluster_neighbors(positions, labels, centers,
                                           max_distance)

    image_clusters = []
    for cluster in clusters:
        image_clusters.append(list(np.take(images, np.array(cluster))))

    save_clusters_with_neighbors(file_path, image_clusters)