test_pointnet_clm.yaml
======================

.. code-block:: yaml
  :linenos:

  # train ae or vae
  #### a template for pcd encoder
  mode: test
  model:
    # can be AE, VAE, SeM and DDPM
    name: ClM
    # encoder config
    encoder:
      name: PointNet
      in_channel: 3
      out_channel: 46
      point_num: 2048
      building_block: dense
      # num_neighbors_k: 20
      local_feature_channels: [64, 64, 128, 256, 512]
      dense_channels: [1024, 512, 256]
      activation: lrelu
      normalization: group
    classification_losses: 
      - name: CE
  loader:
    dataset: 
      name: PcdClsDataset
      data_suffix: .ply
      cls_label: MedPointS
    data_path_list: 
      - path/to/datasets/pcd/MedPointS/classification/fold5
    batch_size: 64
    num_workers: 8
    shuffle: true
    data_transforms:
      - name: Normalize
      - name: FixedPoints
        num: 2048
      - name: ToTensor
        dtype: float
    label_transforms:
      - name: ToOneHot
        num_classes: 47
        ignore_background: true
      - name: ToTensor
        dtype: float
  model_path: path/to/flemme-ckp/MedPointS/PointNet/PointNet_CLS/ckp_best_loss.pth


  evaluation_metrics:
    cls:
      - name: ACC
  tsne_visualization: 
    top_n: 10
    title: t-SNE Visualization of Top-10 Classes
    vis_dim: 2
    label_names: MedPointS