提交 852039b7 编写于 作者: C caojian05

fix dataset deepcopy failed while depipline referenced in the iterator

上级 24fff5fe
......@@ -231,6 +231,9 @@ class Iterator:
def num_classes(self):
return self.depipeline.GetNumClasses()
def __deepcopy__(self, memo):
return Iterator(copy.deepcopy(self.dataset, memo))
class DictIterator(Iterator):
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册