Hugging Face's logo

torch-pointcloud
/
litept-small.structured3d.prs-eth

Safetensors
PyTorch
torch-pointcloud
point-cloud
3d
litept
segmentation
Model card Files
xet
 Community
litept-small.structured3d.prs-eth / config.json
arthurdjn's picture
arthurdjn
Update README.md and config.json and model.safetensors for litept-small.structured3d.prs-eth
20012f4 verified
Raw History Blame Contribute Delete
2.1 kB
{
"library_name": "torch-pointcloud",
"name": "litept-small.structured3d.prs-eth",
"task": "semantic-segmentation",
"hparams": {
"in_channels": 6,
"num_classes": 25,
"serialization_orders": [
"z",
"z-trans",
"hilbert",
"hilbert-trans"
],
"shuffle_serialization_orders": true,
"strides": [
2,
2,
2,
2
],
"encoder_depths": [
2,
2,
2,
6,
2
],
"encoder_channels": [
36,
72,
144,
252,
504
],
"encoder_num_heads": [
2,
4,
8,
14,
28
],
"encoder_patch_size": [
1024,
1024,
1024,
1024,
1024
],
"encoder_conv": [
true,
true,
true,
false,
false
],
"encoder_attention": [
false,
false,
false,
true,
true
],
"decoder_depths": [
0,
0,
0,
0
],
"decoder_channels": [
252,
144,
72,
72
],
"decoder_num_heads": [
14,
8,
4,
4
],
"decoder_patch_size": [
1024,
1024,
1024,
1024
],
"decoder_conv": [
false,
false,
false,
false
],
"decoder_attention": [
false,
false,
false,
false
],
"norm": "batch_norm",
"norm_kwargs": {
"eps": 0.001,
"momentum": 0.01
},
"act": "gelu",
"mlp_ratio": 4.0,
"drop_path": 0.3,
"rope_base": 100.0,
"use_flash_attn": true
},
"input_keys": [
"x",
"pos_grid",
"batch"
],
"dataset": "structured3d",
"classes": [
"wall",
"floor",
"cabinet",
"bed",
"chair",
"sofa",
"table",
"door",
"window",
"picture",
"desk",
"shelves",
"curtain",
"dresser",
"pillow",
"mirror",
"ceiling",
"refrigerator",
"television",
"nightstand",
"sink",
"lamp",
"otherstructure",
"otherfurniture",
"otherprop"
],
"author": "prs-eth",
"license": "MIT"
}