目录

PCT jittor

概述

这是计算机图形学基础课程的PA3作业,基于给定的pct.py baseline实现。

所做调整

数据增强

  • 随机缩放
  • 随机平移
  • 高斯噪声

优化器

改用 Adam。

训练参数

  • n_point 2048
  • epoch 200
  • lr 0.001

依赖

  • Python 3.x
  • jittor
  • numpy

运行

python pct.py --use_cuda 1 --n_points 2048 --batch_size 32 --epochs 200 --num_workers 2 --lr 0.001
关于

A Jittor implementation of Point Cloud Transformer (PCT) for ModelNet40 classification

379.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802032778号