def cnv_w1a1_cifar10(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "cnv-w1a1"
filename = "./bitfiles/Pynq-Z1/cnv-w1a1.bit"
return FINNExampleOverlay(filename, driver_mode,
_cifar10_cnv_io_shape_dict)
def mobilenetv1_w4a4_imagenet(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "mobilenetv1-w4a4"
filename = find_bitfile(model_name, target_platform)
return FINNExampleOverlay(filename, driver_mode,
_imagenet_top5inds_io_shape_dict)
def cnv_w2a2_cifar10(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "cnv-w2a2"
filename = find_bitfile(model_name, target_platform)
return FINNExampleOverlay(filename, driver_mode,
_cifar10_cnv_io_shape_dict)
def vgg10_w4a4_radioml(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "vgg10-radioml-w4a4"
filename = find_bitfile(model_name, target_platform)
fclk_mhz = 250.0
return FINNExampleOverlay(
filename,
driver_mode,
_radioml_io_shape_dict,
fclk_mhz=fclk_mhz,
)
def resnet50_w1a2_imagenet(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "resnet50-w1a2"
filename = find_bitfile(model_name, target_platform)
runtime_weight_dir = find_runtime_weights(model_name, target_platform)
return FINNExampleOverlay(
filename,
driver_mode,
_imagenet_resnet50_top5inds_io_shape_dict,
runtime_weight_dir=runtime_weight_dir,
)
def mobilenetv1_w4a4_imagenet(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "mobilenetv1-w4a4"
filename = find_bitfile(model_name, target_platform)
if target_platform in ["ZCU104"]:
runtime_weight_dir = find_runtime_weights(model_name, target_platform)
else:
runtime_weight_dir = ""
# target 185 MHz for Zynq (this is ignored for Alveo)
fclk_mhz = 185.0
return FINNExampleOverlay(
filename,
driver_mode,
_imagenet_top5inds_io_shape_dict,
runtime_weight_dir=runtime_weight_dir,
fclk_mhz=fclk_mhz,
)
def tfc_w1a1_mnist(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "tfc-w1a1"
filename = "./bitfiles/Pynq-Z1/tfc-w1a1.bit"
return FINNExampleOverlay(filename, driver_mode, _mnist_fc_io_shape_dict)
def tfc_w2a2_mnist(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "tfc-w2a2"
filename = find_bitfile(model_name, target_platform)
return FINNExampleOverlay(filename, driver_mode, _mnist_fc_io_shape_dict)
def bincop_cnv(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "bincop-cnv"
filename = find_bitfile(model_name, target_platform)
return FINNExampleOverlay(filename, driver_mode, _bincop_cnv_io_shape_dict)
def kws_mlp(target_platform=None):
target_platform = resolve_target_platform(target_platform)
driver_mode = get_driver_mode()
model_name = "kwsmlp-w3a3"
filename = find_bitfile(model_name, target_platform)
return FINNExampleOverlay(filename, driver_mode, _gscv2_mlp_io_shape_dict)