Spaces:
Running
on
Zero
Running
on
Zero
File size: 296 Bytes
9e15541 |
1 2 3 4 5 6 7 8 9 10 |
from .semantic_head import SemanticHead
def make_downstream_head(conf):
head_type = conf.get("type", None)
if head_type == "segmentation":
return SemanticHead.from_conf(conf)
else:
raise NotImplementedError(f"Downstream head type '{head_type}' is not implemented.")
|