olib-jet-qc / config.json
ibnbd's picture
Init
88fc31a verified
Raw
History Blame Contribute Delete
1.15 kB
{
"model_type": "ProductClassifier",
"architecture": {
"name": "ProductClassifier",
"input_modalities": [
"depth",
"rgb"
],
"output_classes": 8,
"backbone": "Multi-modal CNN",
"parameters": 629599872,
"input_resolution": [
320,
240
],
"sensor_type": "Opene8008B QVGA ToF"
},
"input_specs": {
"depth_data": {
"shape": "[batch_size, 1, 240, 320]",
"dtype": "float32",
"range": "[0.0, 10.0]",
"unit": "meters"
},
"rgb_data": {
"shape": "[batch_size, 3, 240, 320]",
"dtype": "float32",
"range": "[0.0, 1.0]",
"format": "RGB"
}
},
"output_specs": {
"logits": {
"shape": "[batch_size, num_classes]",
"dtype": "float32",
"description": "Raw classification logits"
},
"probabilities": {
"shape": "[batch_size, num_classes]",
"dtype": "float32",
"description": "Softmax probabilities"
}
},
"hardware": {
"target_device": "NVIDIA Jetson AGX Orin",
"min_memory_gb": 8,
"cuda_required": true,
"jetpack_version": "5.1+",
"python_version": "3.8+"
}
}