文件名称：point-transformer-pytorch:Pytorch中Point Transformer层的实现
文件大小：27KB
文件格式：ZIP
更新时间：2024-05-02 06:50:34
deep-learning point-cloud artificial-intelligence attention-mechanism Python
点变压器-火炬 在Pytorch中实现自注意层。 上面的简单电路似乎允许他们的小组在点云分类和分割方面胜过所有以前的方法。 安装 $ pip install point-transformer-pytorch 用法 import torch from point_transformer_pytorch import PointTransformerLayer attn = PointTransformerLayer ( dim = 128 , pos_mlp_hidden_dim = 64 , attn_mlp_hidden_mult = 4 ) feats = torch . randn ( 1 , 16 , 128 ) pos = torch . randn ( 1 , 16 , 3 ) mask = torch . ones ( 1 , 16 ). bool
【文件预览】：
point-transformer-pytorch-main
----point_transformer_pytorch()
--------__init__.py(86B)
--------point_transformer_pytorch.py(3KB)
----point-transformer-layer.png(21KB)
----.github()
--------workflows()
----LICENSE(1KB)
----setup.py(773B)
----.gitignore(2KB)
----README.md(2KB)