Sonata / processor_config.json
bigshanedogg's picture
Upload Sonata (transformers port, original code)
c298327 verified
Raw History Blame Contribute Delete
1 kB
{
"auto_map": {
"AutoProcessor": "processing_sonata.SonataProcessor"
},
"feature_extractor": {
"auto_map": {
"AutoFeatureExtractor": "feature_extraction_sonata.SonataFeatureExtractor",
"AutoProcessor": "processing_sonata.SonataProcessor"
},
"feature_extractor_type": "SonataFeatureExtractor",
"transform": [
{
"apply_z": true,
"type": "CenterShift"
},
{
"grid_size": 0.02,
"hash_type": "fnv",
"mode": "train",
"return_grid_coord": true,
"return_inverse": true,
"type": "GridSample"
},
{
"type": "NormalizeColor"
},
{
"type": "ToTensor"
},
{
"feat_keys": [
"coord",
"color",
"normal"
],
"keys": [
"coord",
"grid_coord",
"color",
"inverse"
],
"type": "Collect"
}
]
},
"processor_class": "SonataProcessor"
}