Jelajahi Sumber

fix in InceptionV3 model after the refactoring

Dimitri Korsch 4 tahun lalu
induk
melakukan
844b9fdfc2

+ 0 - 1
cvmodelz/models/base.py

@@ -19,7 +19,6 @@ class BaseModel(abc.ABC):
 		with self.init_scope():
 			self.pool = pooling
 
-
 		if input_size is not None:
 			self.meta.input_size = input_size
 

+ 1 - 0
cvmodelz/models/pretrained/inception/inception_v3.py

@@ -42,6 +42,7 @@ class InceptionV3(PretrainedModelMixin, chainer.Chain):
 		if pretrained_model is not None and isfile(pretrained_model):
 			self.load(pretrained_model, strict=True)
 
+	def init_model_info(self):
 		self.meta = ModelInfo(
 			name="InceptionV3",
 			input_size=299,

+ 0 - 19
cvmodelz/utils/__init__.py

@@ -41,7 +41,6 @@ def print_model_info(model, file=sys.stdout, input_size=None, input_var=None):
 	if name is None:
 		name = model.__class__.__name__
 
-
 	rows = []
 	default_size = 224
 	if hasattr(model, "meta"):
@@ -73,21 +72,3 @@ def print_model_info(model, file=sys.stdout, input_size=None, input_var=None):
 		headers=["Link name", "Input", "Output"],
 		tablefmt="fancy_grid"))
 
-
-if __name__ == "__main__":
-	class Foo:
-		attr = None
-		def __str__(self):
-			return f"<{hex(id(self))}: attr={self.attr}>"
-
-
-	o = Foo()
-	o.attr = Foo()
-	o.attr.attr = Foo()
-
-	print(o,
-		get_attr_from_path(o, "attr"),
-		get_attr_from_path(o, "attr/"),
-		get_attr_from_path(o, "attr/attr"),
-		sep="\n"
-	)