All notable changes to this project will be documented in this file.
The format is based on Keep a Changelog, and this project adheres to Semantic Versioning.
- Support for the IRALab benchmark (https://arxiv.org/abs/2003.12841), with data from the ETH, Canadian Planetary, Kaist and TUM datasets.
- Dataset configurations are saved in the checkpoints so that models can be created without requiring the actual dataset
- BatchNorm1d fix
- More general API for Minkowski with support for Bottleneck blocks and Squeeze and excite.
- Teaser support for registration
- Examples for using pretrained registration models
- Pointnet2 forward examples for classification, segmentation
- S3DIS automatic download and panoptic support and cylinder sampling
- Moved to PyTorch 1.6 as officialy supported PyTorch version
- Add
context = ssl._create_unverified_context()
,data = urllib.request.urlopen(url, context=context)
withindownload_ulr
, so ModelNet and ShapeNet can download.
- Support scannet test dataset and automatic generation of submission files using the eval.py script
- Full res predictions on Scannet with voting
- VoteNet model and losses
- Tracker for object detection
- Models can specify which attributes they need from the data in order to forward and train properly
- Full res predictions on ShapeNet with voting
- Trainer class to handle train / eval
- Add testing for Trainer:
- Segmentation: PointNet2 on cap ShapeNet
- Segmentation: KPConv on scannetV2
- Object Detection: VoteNet on scannetV2
- Add VoteNet Paper / Backbones within API
- Windows support
- Weights are uploaded to wandb at the end of the run
- Added PointGroup https://arxiv.org/pdf/2007.01294.pdf
- Added PretrainedRegistry allowing model weight to be downloaded directly from wandb and DatasetMocking
- Added script for s3dis cross-validation [scripts/cv_s3dis.py]. 6 different pretrained models will be downloaded, evaluated on full resolution and confusion matrice will be summed to get all metrics.
- mAP tracker for Panoptic segmentation
- evaluation output folder is now a subfolder of the checkpoint it uses
- saves model checkpoints to wandb
- GridSampling3D now creates a new attribute
coords
that stores the non quantized position when the transform is called inquantize
mode - cuda parameter can be given in command line to select the GPU to use
- Updated to pytorch geometric 1.6.0
- LR secheduler resume is broken for update on batch number #328
- ElasticDistortion transform is now fully functional
- We now support the latest PyTorch
- Migration to the latest PyTorch Geometric and dependencies
- #273 (support python 3.7)
- Pre transform is being overriden by the inference transform
- Customizable number of channels at the output of the API models
- API models expose output number of channels as a property
- Added Encoder to the API
- Sampled ModelNet dataset for point clouds