pointnext-official 0.1.0__tar.gz

pointnext_official-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+    MIT License
+
+    Copyright (c) Guocheng Qian.
+
+    Permission is hereby granted, free of charge, to any person obtaining a copy
+    of this software and associated documentation files (the "Software"), to deal
+    in the Software without restriction, including without limitation the rights
+    to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+    copies of the Software, and to permit persons to whom the Software is
+    furnished to do so, subject to the following conditions:
+
+    The above copyright notice and this permission notice shall be included in all
+    copies or substantial portions of the Software.
+
+    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+    IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+    AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+    LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+    OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+    SOFTWARE

pointnext_official-0.1.0/MANIFEST.in

@@ -0,0 +1,20 @@
+include LICENSE
+include README.md
+include requirements.txt
+include install.sh
+include update.sh
+recursive-include cfgs *.yaml
+recursive-include docs *.md *.png *.jpg *.jpeg
+recursive-include examples *.py
+recursive-include tools *.py
+recursive-include notebooks *.ipynb
+recursive-include metadata *.md *.sha256
+prune openpoints/.git
+prune .git
+prune data
+prune log
+prune logs
+prune wandb
+prune dist
+prune build
+prune *.egg-info

pointnext_official-0.1.0/PKG-INFO

@@ -0,0 +1,174 @@
+Metadata-Version: 2.4
+Name: pointnext_official
+Version: 0.1.0
+Summary: PointNeXt release helpers, checkpoint download utilities, and metadata for pip-installed OpenPoints/PointNeXt users.
+Author: Guocheng Qian
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/guochengqian/PointNeXt
+Project-URL: Repository, https://github.com/guochengqian/PointNeXt
+Project-URL: Documentation, https://guochengqian.github.io/PointNeXt/
+Project-URL: Issues, https://github.com/guochengqian/PointNeXt/issues
+Project-URL: Checkpoints, https://huggingface.co/guochengqian/pointnext
+Keywords: point cloud,deep learning,pytorch,PointNeXt,OpenPoints
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openpoints>=0.1.0
+Requires-Dist: huggingface_hub>=0.19
+Requires-Dist: PyYAML>=5.4
+Provides-Extra: dev
+Requires-Dist: build>=1; extra == "dev"
+Requires-Dist: twine>=4; extra == "dev"
+Requires-Dist: pytest>=7; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: mkdocs-material; extra == "docs"
+Requires-Dist: mkdocs-awesome-pages-plugin; extra == "docs"
+Requires-Dist: mdx_truly_sane_lists; extra == "docs"
+Dynamic: license-file
+
+# PointNeXt
+
+### [[arXiv]](https://arxiv.org/abs/2206.04670) | [[OpenPoints Library]](https://github.com/guochengqian/openpoints) | [[Online Documentation]](https://guochengqian.github.io/PointNeXt/)
+
+<p align="center">
+<img src="docs/projects/misc/effects_training_scaling.png" width=85% height=85% class="center">
+</p>
+
+Official PyTorch implementation for the following paper:
+
+**PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies**
+
+*by [Guocheng Qian](https://www.gcqian.com/), [Yuchen Li](https://cemse.kaust.edu.sa/vision-cair/people/person/yuchen-li), [Houwen Peng](https://houwenpeng.com/), [Jinjie Mai](https://cemse.kaust.edu.sa/people/person/jinjie-mai), [Hasan Hammoud](https://cemse.kaust.edu.sa/ece/people/person/hasan-abed-al-kader-hammoud), [Mohamed Elhoseiny](http://www.mohamed-elhoseiny.com/), [Bernard Ghanem](https://www.bernardghanem.com/)*
+
+**TL;DR:** We propose improved training and model scaling strategies to boost PointNet++ to the state-of-the-art level. PointNet++ with the proposed model scaling is named as PointNeXt, the next version of PointNets. 
+
+
+<p align="center">
+<img src="docs/projects/misc/pointnext.jpeg" width=85% height=85% class="center">
+</p>
+
+## News
+-  :boom: Sep, 2022: [**PointNeXt**](https://arxiv.org/pdf/2206.04670v1.pdf) accepted by NeurIPS'22
+-  :boom: Jun, 2022: Code released
+
+
+## Features
+In the PointNeXt project, we propose a new and flexible codebase for point-based methods, namely [**OpenPoints**](https://github.com/guochengqian/openpoints). The biggest difference between OpenPoints and other libraries is that we focus more on reproducibility and fair benchmarking. 
+
+1. **Extensibility**: supports many representative networks for point cloud understanding, such as *PointNet, DGCNN, DeepGCN, PointNet++, ASSANet, PointMLP*, and our ***PointNeXt***. More networks can be built easily based on our framework since OpenPoints support a wide range of basic operations including graph convolutions, self-attention, farthest point sampling, ball query, *e.t.c*.
+
+2. **Reproducibility**: all implemented models are trained on various tasks at least three times. Mean±std is provided in the [PointNeXt paper](https://arxiv.org/abs/2206.04670).  *Pretrained models and logs* are available.
+
+3. **Fair Benchmarking**: in PointNeXt, we find a large part of performance gain is due to the training strategies. In OpenPoints, all models are trained with the improved training strategies and all achieve much higher accuracy than the original reported value. 
+
+4. **Ease of Use**: *Build* model, optimizer, scheduler, loss function,  and data loader *easily from cfg*. Train and validate different models on various tasks by simply changing the `cfg\*\*.yaml` file. 
+
+   ```
+   model = build_model_from_cfg(cfg.model)
+   criterion = build_criterion_from_cfg(cfg.criterion_args)
+   ```
+   Here is an example of `pointnet.yaml` (model configuration for PointNet model):
+   ```python
+   model:
+     NAME: BaseCls
+     encoder_args:
+       NAME: PointNetEncoder
+       in_channels: 4
+     cls_args:
+       NAME: ClsHead
+       num_classes: 15
+       in_channels: 1024
+       mlps: [512,256]
+       norm_args: 
+         norm: 'bn1d'
+   ```
+
+5. **Online logging**: *Support [wandb](https://wandb.ai/)* for checking your results anytime anywhere. Just set `wandb.use_wandb=True` in your command.  
+
+   ![docs/misc/wandb.png](docs/misc/wandb.png)
+
+---
+
+## Installation
+
+### Pip packages
+
+The Python libraries are released as installable packages:
+
+```bash
+pip install pointnext_official
+```
+
+`pointnext_official` installs `openpoints` as its core library dependency and provides PointNeXt release metadata and checkpoint download helpers, for example:
+
+```bash
+pointnext-download --list
+pointnext-download modelnet40-pointnext-s-c64 --output-dir ./hf_cache
+```
+
+The PyPI packages are importable without compiling CUDA extensions. Full training/evaluation still requires the custom CUDA/C++ ops, so use a source checkout for benchmark reproduction.
+
+### Source install for training/evaluation
+
+```bash
+git clone --recurse-submodules https://github.com/guochengqian/PointNeXt.git
+cd PointNeXt
+git submodule update --init --recursive
+source update.sh
+source install.sh
+```
+
+If SSH is configured, `git@github.com:guochengqian/PointNeXt.git` also works. CUDA 11.3 was used for the original release. Modify `install.sh` if a different CUDA/PyTorch version is used. See [Install](docs/index.md), [FAQ](docs/faq.md), and [Checkpoints](docs/checkpoints.md) for details.
+
+
+
+## Usage 
+Check our [online documentation](https://guochengqian.github.io/PointNeXt/) for detailed instructions. 
+
+A short instruction: all experiments follow the simple rule to train and test: 
+
+```
+CUDA_VISIBLE_DEVICES=$GPUs python examples/$task_folder/main.py --cfg $cfg $kwargs
+```
+- $GPUs is the list of GPUs to use, for most experiments (ScanObjectNN, ModelNet40, S3DIS), we only use 1 A100 (GPUs=0)
+- $task_folder is the folder name of the experiment. For example, for s3dis segmentation, $task_folder=s3dis
+- $cfg is the path to cfg, for example, s3dis segmentation, $cfg=cfgs/s3dis/pointnext-s.yaml
+- $kwargs are the other keyword arguments to use. For example, testing in S3DIS area 5, $kwargs should be `mode=test, --pretrained_path $pretrained_path`. 
+
+
+## Model Zoo (pretrained weights)
+See [Model Zoo](https://guochengqian.github.io/PointNeXt/modelzoo/) and [checkpoint download docs](docs/checkpoints.md). The recommended new release path hosts large checkpoints and checksum manifests on Hugging Face Hub, while GitHub Releases/PyPI host source and Python packages.
+
+### Visualization
+More examples are available in the [paper](https://arxiv.org/abs/2206.04670). 
+
+![s3dis](docs/projects/misc/s3dis_vis.png)
+![shapenetpart](docs/projects/misc/shapenetpart_vis.png)
+
+---
+
+### Acknowledgment
+This library is inspired by [PyTorch-image-models](https://github.com/rwightman/pytorch-image-models) and [mmcv](https://github.com/open-mmlab/mmcv). 
+
+
+
+### Citation
+If you find PointNeXt or the OpenPoints codebase useful, please cite:
+```tex
+@InProceedings{qian2022pointnext,
+  title   = {PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies},
+  author  = {Qian, Guocheng and Li, Yuchen and Peng, Houwen and Mai, Jinjie and Hammoud, Hasan and Elhoseiny, Mohamed and Ghanem, Bernard},
+  booktitle=Advances in Neural Information Processing Systems (NeurIPS),
+  year    = {2022},
+}
+```

pointnext_official-0.1.0/README.md

@@ -0,0 +1,136 @@
+# PointNeXt
+
+### [[arXiv]](https://arxiv.org/abs/2206.04670) | [[OpenPoints Library]](https://github.com/guochengqian/openpoints) | [[Online Documentation]](https://guochengqian.github.io/PointNeXt/)
+
+<p align="center">
+<img src="docs/projects/misc/effects_training_scaling.png" width=85% height=85% class="center">
+</p>
+
+Official PyTorch implementation for the following paper:
+
+**PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies**
+
+*by [Guocheng Qian](https://www.gcqian.com/), [Yuchen Li](https://cemse.kaust.edu.sa/vision-cair/people/person/yuchen-li), [Houwen Peng](https://houwenpeng.com/), [Jinjie Mai](https://cemse.kaust.edu.sa/people/person/jinjie-mai), [Hasan Hammoud](https://cemse.kaust.edu.sa/ece/people/person/hasan-abed-al-kader-hammoud), [Mohamed Elhoseiny](http://www.mohamed-elhoseiny.com/), [Bernard Ghanem](https://www.bernardghanem.com/)*
+
+**TL;DR:** We propose improved training and model scaling strategies to boost PointNet++ to the state-of-the-art level. PointNet++ with the proposed model scaling is named as PointNeXt, the next version of PointNets. 
+
+
+<p align="center">
+<img src="docs/projects/misc/pointnext.jpeg" width=85% height=85% class="center">
+</p>
+
+## News
+-  :boom: Sep, 2022: [**PointNeXt**](https://arxiv.org/pdf/2206.04670v1.pdf) accepted by NeurIPS'22
+-  :boom: Jun, 2022: Code released
+
+
+## Features
+In the PointNeXt project, we propose a new and flexible codebase for point-based methods, namely [**OpenPoints**](https://github.com/guochengqian/openpoints). The biggest difference between OpenPoints and other libraries is that we focus more on reproducibility and fair benchmarking. 
+
+1. **Extensibility**: supports many representative networks for point cloud understanding, such as *PointNet, DGCNN, DeepGCN, PointNet++, ASSANet, PointMLP*, and our ***PointNeXt***. More networks can be built easily based on our framework since OpenPoints support a wide range of basic operations including graph convolutions, self-attention, farthest point sampling, ball query, *e.t.c*.
+
+2. **Reproducibility**: all implemented models are trained on various tasks at least three times. Mean±std is provided in the [PointNeXt paper](https://arxiv.org/abs/2206.04670).  *Pretrained models and logs* are available.
+
+3. **Fair Benchmarking**: in PointNeXt, we find a large part of performance gain is due to the training strategies. In OpenPoints, all models are trained with the improved training strategies and all achieve much higher accuracy than the original reported value. 
+
+4. **Ease of Use**: *Build* model, optimizer, scheduler, loss function,  and data loader *easily from cfg*. Train and validate different models on various tasks by simply changing the `cfg\*\*.yaml` file. 
+
+   ```
+   model = build_model_from_cfg(cfg.model)
+   criterion = build_criterion_from_cfg(cfg.criterion_args)
+   ```
+   Here is an example of `pointnet.yaml` (model configuration for PointNet model):
+   ```python
+   model:
+     NAME: BaseCls
+     encoder_args:
+       NAME: PointNetEncoder
+       in_channels: 4
+     cls_args:
+       NAME: ClsHead
+       num_classes: 15
+       in_channels: 1024
+       mlps: [512,256]
+       norm_args: 
+         norm: 'bn1d'
+   ```
+
+5. **Online logging**: *Support [wandb](https://wandb.ai/)* for checking your results anytime anywhere. Just set `wandb.use_wandb=True` in your command.  
+
+   ![docs/misc/wandb.png](docs/misc/wandb.png)
+
+---
+
+## Installation
+
+### Pip packages
+
+The Python libraries are released as installable packages:
+
+```bash
+pip install pointnext_official
+```
+
+`pointnext_official` installs `openpoints` as its core library dependency and provides PointNeXt release metadata and checkpoint download helpers, for example:
+
+```bash
+pointnext-download --list
+pointnext-download modelnet40-pointnext-s-c64 --output-dir ./hf_cache
+```
+
+The PyPI packages are importable without compiling CUDA extensions. Full training/evaluation still requires the custom CUDA/C++ ops, so use a source checkout for benchmark reproduction.
+
+### Source install for training/evaluation
+
+```bash
+git clone --recurse-submodules https://github.com/guochengqian/PointNeXt.git
+cd PointNeXt
+git submodule update --init --recursive
+source update.sh
+source install.sh
+```
+
+If SSH is configured, `git@github.com:guochengqian/PointNeXt.git` also works. CUDA 11.3 was used for the original release. Modify `install.sh` if a different CUDA/PyTorch version is used. See [Install](docs/index.md), [FAQ](docs/faq.md), and [Checkpoints](docs/checkpoints.md) for details.
+
+
+
+## Usage 
+Check our [online documentation](https://guochengqian.github.io/PointNeXt/) for detailed instructions. 
+
+A short instruction: all experiments follow the simple rule to train and test: 
+
+```
+CUDA_VISIBLE_DEVICES=$GPUs python examples/$task_folder/main.py --cfg $cfg $kwargs
+```
+- $GPUs is the list of GPUs to use, for most experiments (ScanObjectNN, ModelNet40, S3DIS), we only use 1 A100 (GPUs=0)
+- $task_folder is the folder name of the experiment. For example, for s3dis segmentation, $task_folder=s3dis
+- $cfg is the path to cfg, for example, s3dis segmentation, $cfg=cfgs/s3dis/pointnext-s.yaml
+- $kwargs are the other keyword arguments to use. For example, testing in S3DIS area 5, $kwargs should be `mode=test, --pretrained_path $pretrained_path`. 
+
+
+## Model Zoo (pretrained weights)
+See [Model Zoo](https://guochengqian.github.io/PointNeXt/modelzoo/) and [checkpoint download docs](docs/checkpoints.md). The recommended new release path hosts large checkpoints and checksum manifests on Hugging Face Hub, while GitHub Releases/PyPI host source and Python packages.
+
+### Visualization
+More examples are available in the [paper](https://arxiv.org/abs/2206.04670). 
+
+![s3dis](docs/projects/misc/s3dis_vis.png)
+![shapenetpart](docs/projects/misc/shapenetpart_vis.png)
+
+---
+
+### Acknowledgment
+This library is inspired by [PyTorch-image-models](https://github.com/rwightman/pytorch-image-models) and [mmcv](https://github.com/open-mmlab/mmcv). 
+
+
+
+### Citation
+If you find PointNeXt or the OpenPoints codebase useful, please cite:
+```tex
+@InProceedings{qian2022pointnext,
+  title   = {PointNeXt: Revisiting PointNet++ with Improved Training and Scaling Strategies},
+  author  = {Qian, Guocheng and Li, Yuchen and Peng, Houwen and Mai, Jinjie and Hammoud, Hasan and Elhoseiny, Mohamed and Ghanem, Bernard},
+  booktitle=Advances in Neural Information Processing Systems (NeurIPS),
+  year    = {2022},
+}
+```

pointnext_official-0.1.0/cfgs/default.yaml

@@ -0,0 +1,70 @@
+# distributed
+dist_url: tcp://localhost:8888
+dist_backend: 'nccl'
+multiprocessing_distributed: False
+ngpus_per_node: 1
+world_size: 1
+launcher: 'mp'
+local_rank: 0
+
+use_gpu: True
+seed: null 
+
+# ---------------------------------------------------------------------------- #
+# Training cfgs
+# ---------------------------------------------------------------------------- #
+epoch: 0
+epochs: 100 
+
+ignore_index: null
+val_fn: validate
+deterministic: False
+sync_bn: False
+
+criterion_args:
+  NAME: CrossEntropy
+use_mask: False # in some datasets, requiring data mask in input
+
+grad_norm_clip: null
+layer_decay: 0 # no layer decay by default
+
+step_per_update: 1
+start_epoch: 1 
+sched_on_epoch: True
+
+# We support wandb for online results collection, please check their documentation for details: https://docs.wandb.ai/
+wandb:
+  use_wandb: False 
+
+use_amp: False
+# ---------------------------------------------------------------------------- #
+# Evaluation cfgs
+# ---------------------------------------------------------------------------- #
+use_voting: False # we do not use any voting. Voting is not a good practice 
+val_freq: 10
+
+# ---------------------------------------------------------------------------- #
+# io and misc
+# ---------------------------------------------------------------------------- #
+resume: False
+test: False 
+finetune: False 
+
+mode: train # set to test in evaluation only mode
+logname: null
+load_path: null
+
+print_freq: 50 
+save_freq: -1 # saving frequency for ckpt. -1 only saving the latest and the best.
+
+root_dir: log/
+pretrained_path: null
+
+
+# ---------------------------------------------------------------------------- #
+# data
+# ---------------------------------------------------------------------------- #
+datatransforms:
+  train: null
+  val: null
+feature_keys: pos

pointnext_official-0.1.0/cfgs/modelnet40ply2048/assanet-l.yaml

@@ -0,0 +1,43 @@
+model:
+  NAME: BaseCls
+  encoder_args:
+    NAME: PointNet2Encoder
+    in_channels: 3 
+    strides: [2, 2, 2, 2]
+    blocks: [3, 3, 3, 3]
+    width: 128
+    width_scaling: 3
+    double_last_channel: False
+    layers: 3
+    use_res: True 
+    query_as_support: True
+    mlps: null 
+    stem_conv: True
+    stem_aggr: True
+    radius: 0.15
+    radius_scaling: 1.5
+    block_radius_scaling: 1.5 
+    num_samples: [[20, 36], [20, 36], [20, 36], [20, 36]] 
+    sampler: fps
+    aggr_args:
+      NAME: 'ASSA'
+      feature_type: 'assa'
+      anisotropic: True 
+      reduction: 'mean'
+    group_args:
+      NAME: 'ballquery'
+      use_xyz: True
+      normalize_dp: True
+    conv_args:
+      order: conv-norm-act
+    act_args:
+      act: 'relu'
+    norm_args:
+      norm: 'bn'
+  cls_args: 
+    NAME: ClsHead
+    num_classes: 40
+    global_feat: max,avg
+    mlps: [512, 256]
+    norm_args: 
+      norm: 'bn1d'

pointnext_official-0.1.0/cfgs/modelnet40ply2048/deepgcn.yaml

@@ -0,0 +1,44 @@
+# GFLOPs  GMACs   Params.(M)
+#  3.88    1.93    2.220
+# Throughput (ins./s): 262.8308528647436
+
+model: 
+  NAME: BaseCls
+  encoder_args:
+    NAME: DeepGCN
+    in_channels: 3
+    channels: 64
+    n_classes: 40 
+    emb_dims: 1024
+    n_blocks: 14
+    conv: 'edge'
+    block: 'res'
+    k: 9
+    epsilon: 0.2
+    use_stochastic: True
+    use_dilation: True 
+    dropout: 0.5
+    norm_args: {'norm': 'bn'}
+    act_args: {'act': 'relu'}
+  cls_args: 
+    NAME: ClsHead
+    num_classes: 40
+    mlps: [512, 256]
+    norm_args: 
+      norm: 'bn1d'
+lr:
+  0.1
+optimizer:
+  NAME: momentum 
+  weight_decay: 1.0e-4
+  momentum: 0.9
+  filter_bias_and_bn: False
+
+# scheduler
+sched: cosine
+epochs: 400
+warmup_epochs:  0
+warmup_lr: 1.0e-6
+min_lr: 0.001
+
+sync_bn: False

pointnext_official-0.1.0/cfgs/modelnet40ply2048/default.yaml

@@ -0,0 +1,67 @@
+# Dataset Related
+num_points: 1024  # number of poins actually used in training and evaluation
+dataset:
+  common:
+    NAME: ModelNet40Ply2048
+    data_dir: './data/ModelNet40Ply2048'
+  train:
+    split: train
+    num_points: 1024  # in training, use sampled 1024 points for data augmentation. 
+  val:
+    split: test
+    num_points: 1024  # in testing, use uniformly pre-sampled 1024 points for evaluation (following https://github.com/lulutang0608/Point-BERT) 
+
+feature_keys: pos
+
+datatransforms:
+  train: [PointsToTensor, PointCloudScaleAndTranslate]  # rotation does not help
+  vote: [PointCloudScaleAndTranslate]
+  val: [PointsToTensor]
+  kwargs:
+    shift: [0.2, 0.2, 0.2]
+batch_size: 32
+dataloader:
+  num_workers: 6
+
+num_classes: 40
+# ---------------------------------------------------------------------------- #
+# Training cfgs
+# ---------------------------------------------------------------------------- #
+# training receipe borrowed from: https://github.com/yanx27/Pointnet_Pointnet2_pytorch
+
+# ---------------------------------------------------------------------------- #
+# Training cfgs
+# ---------------------------------------------------------------------------- #
+# this one is better. 
+sched: cosine
+epochs: 600
+warmup_epochs: 0
+min_lr: null 
+
+# Training parameters
+lr: 0.001
+optimizer:
+ NAME: 'adamw'
+ weight_decay: 0.05 
+
+grad_norm_clip: 1
+
+criterion_args:
+  NAME: SmoothCrossEntropy
+  label_smoothing: 0.2
+
+# ---------------------------------------------------------------------------- #
+# io and misc
+# ---------------------------------------------------------------------------- #
+log_dir: 'modelnet40'
+print_freq: 10
+val_freq: 1
+
+# ----------------- Model related
+val_batch_size: 64
+pretrained_path: null 
+
+wandb:
+  project: PointNeXt-ModelNet40Ply2048
+
+seed: null

pointnext_official-0.1.0/cfgs/modelnet40ply2048/pix4point.yaml

@@ -0,0 +1,62 @@
+model:
+  NAME: BaseCls
+  encoder_args:
+    NAME: PointViT
+    in_channels: 3 
+    embed_dim: 384
+    depth: 12
+    num_heads: 6
+    mlp_ratio: 4.
+    drop_rate: 0.
+    attn_drop_rate: 0.1
+    drop_path_rate: 0.0
+    qkv_bias: True
+    act_args:
+      act: 'gelu'
+    norm_args:
+      norm: 'ln'
+      eps: 1.0e-6
+    embed_args:
+      NAME: PointPatchEmbed
+      feature_type: 'dp'
+      reduction: 'max'
+      sample_ratio: 0.25  # ablation: 0.25
+      normalize_dp: False 
+      group_size: 32
+      subsample: 'fps'
+      group: 'knn'
+      norm_args: bn # ablation: in
+      conv_args:
+        order: conv-norm-act
+      layers: 4
+      channels: [128, 256, 512]
+    global_feat: cls,max,avg
+  cls_args: 
+    NAME: ClsHead 
+    num_classes: 40
+    mlps: [256, 256]
+    norm_args: 
+      norm: 'bn1d'
+
+pretrained_path: pretrained/imagenet/small_21k_224.pth
+mode: finetune_encoder
+
+
+criterion_args:
+  NAME: SmoothCrossEntropy
+  label_smoothing: 0.2
+
+# Optimizer
+lr: 5.0e-4 
+optimizer:
+ NAME: 'adamw'
+ weight_decay: 0.05
+
+# scheduler
+epochs: 300
+sched: cosine 
+warmup_epochs: 0
+min_lr: 1.0e-6 
+
+wandb:
+  project: pix4point-modelnet40

pointnext_official-0.1.0/cfgs/modelnet40ply2048/pointmlp.yaml

@@ -0,0 +1,46 @@
+# GFLOPs  GMACs   Params.(M)
+#  31.36   15.61   13.239
+model:
+  NAME: PointMLP
+  in_channels: 3
+  points: 1024
+  num_classes: 40
+  embed_dim: 64
+  groups: 1
+  res_expansion: 1.0
+  activation: "relu"
+  bias: False
+  use_xyz: False
+  normalize: "anchor"
+  dim_expansion: [ 2, 2, 2, 2 ]
+  pre_blocks: [ 2, 2, 2, 2 ]
+  pos_blocks: [ 2, 2, 2, 2 ]
+  k_neighbors: [ 24, 24, 24, 24 ]
+  reducers: [ 2, 2, 2, 2 ]
+
+lr:
+  0.1
+optimizer:
+  NAME: 'momentum'
+  momentum: 0.9
+  weight_decay: 2.0e-4
+
+sched: cosine
+epochs: 300
+t_max: 300
+min_lr: 0.005
+warmup_epochs: 0  # Later change to 10?
+
+criterion_args:
+  NAME: SmoothCrossEntropy
+  label_smoothing: 0.2
+
+datatransforms:
+  train: [PointsToTensor, PointCloudScaleAndTranslate]
+  vote: [PointCloudScaleAndTranslate]
+  val: [PointsToTensor]
+  kwargs:
+    shift: [0.2, 0.2, 0.2]
+
+batch_size: 32
+val_batch_size: 64