File size: 605 Bytes
4dce182 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from .common import count_params
from .detector import Detector
from .extractor import Extractor
from .orientation import OrientationNet
from .recognizer import Recognizer
BUDGET = {"detector": 1_200_000, "recognizer": 3_200_000, "extractor": 400_000, "orientation": 200_000}
BUILDERS = {"detector": Detector, "recognizer": Recognizer, "extractor": Extractor, "orientation": OrientationNet}
def build(name, **kwargs):
return BUILDERS[name](**kwargs)
def parameter_report():
rows = {name: count_params(build(name)) for name in BUILDERS}
rows["total"] = sum(rows.values())
return rows
|