arthurdjn's picture
Update README.md and config.json and model.safetensors for pointnet.modelnet40.jiawei-ren
5a8d19d verified
Raw History Blame Contribute Delete
1.27 kB
{
"library_name": "torch-pointcloud",
"name": "pointnet.modelnet40.jiawei-ren",
"task": "classification",
"hparams": {
"in_channels": 0,
"num_classes": 40,
"dropout": 0.3,
"mlp1_channels": [
64
],
"mlp2_channels": [
128,
1024
],
"tnet_mlp1_dims": [
64,
128,
1024
],
"tnet_mlp2_dims": [
512,
256
],
"head_channels": [
512,
256
],
"act_first": false,
"act_last": false
},
"input_keys": [
"x",
"pos",
"batch"
],
"dataset": "modelnet40",
"metrics": {
"OA": 90.68,
"mAcc": 86.54
},
"classes": [
"airplane",
"bathtub",
"bed",
"bench",
"bookshelf",
"bottle",
"bowl",
"car",
"chair",
"cone",
"cup",
"curtain",
"desk",
"door",
"dresser",
"flower_pot",
"glass_box",
"guitar",
"keyboard",
"lamp",
"laptop",
"mantel",
"monitor",
"night_stand",
"person",
"piano",
"plant",
"radio",
"range_hood",
"sink",
"sofa",
"stairs",
"stool",
"table",
"tent",
"toilet",
"tv_stand",
"vase",
"wardrobe",
"xbox"
],
"author": "jiawei-ren",
"license": "Apache-2.0"
}