Skip to content

PRBonn/shape_completion_toolkit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

45c4ab3 · Jun 4, 2024

History

3 Commits
Jun 4, 2024
Jun 3, 2024
Jun 3, 2024
Jun 3, 2024
Jun 3, 2024

Repository files navigation

Shape Completion Toolkit

To speed up the usage of our dataset, we provide a development kit providing a PyTorch-based data loader and a small library for computing the metrics that we use in the challenge relative to this dataset.

Dataloader

from torch.utils.data import DataLoader
from dataloader import ShapeCompletionDataset

shape_completion_dataset = ShapeCompletionDataset(data_source='path_to_data/')
dataloader = DataLoader(shape_completion_dataset)

for item in dataloader:
    <do stuff>

Metrics

import open3d as o3d

from chamfer_distance import ChamferDistance
from precision_recall import PrecisionRecall

cd = ChamferDistance()
pr = PrecisionRecall(min_t=0.001,max_t=0.01,num=10)

# list of prediction and corresponding groundtruth
predictions = [...]
groundtruths = [...]

for prediction, groundtruth in zip(predictions,groundtruths):

    cd.update(groundtruth, prediction)  
    pr.update(groundtruth, prediction)  

final_cd_metric = cd.compute()
final_pr_metrics = pr.compute()

cd.reset()
pr.reset()

About

ECCV 2024 Shape Completion Competition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages