Spaces:

hasibzunair
/

fifa-tryon-demo

Runtime error

App Files Files Community

fifa-tryon-demo / Self-Correction-Human-Parsing-for-ACGPN /networks /__init__.py

hasibzunair's picture

added files

4a285f6 over 1 year ago

raw history blame contribute delete

No virus

311 Bytes

	from __future__ import absolute_import

	from networks.AugmentCE2P import resnet101

	__factory = {
	'resnet101': resnet101,
	}


	def init_model(name, args, *kwargs):
	if name not in __factory.keys():
	raise KeyError("Unknown model arch: {}".format(name))
	return __factory[name](args, *kwargs)