sandbox338's picture
Add model configuration
26ea88e verified
raw
history blame contribute delete
373 Bytes
{
"model_type": "detectron2-faster-rcnn",
"architecture": "OptimizedInceptionWrapper",
"num_classes": 10,
"class_names": [
"antelope",
"buffalo",
"elephant",
"giraffe",
"gorilla",
"leopard",
"lion",
"rhino",
"wolf",
"zebra"
],
"confidence_threshold": 0.5,
"nms_threshold": 0.3,
"input_size": [
800,
1333
]
}