def worker_loop(self, worker_num):
self.worker_num = worker_num
with open(self.args.classifier, "r") as fh:
print "unpickling"
self.classifier = cPickle.load(fh)
print "done"
ParallelMapper.worker_loop(self, worker_num)
def __init__(self):
ParallelMapper.__init__(self, ["bmp"],
[comm.MultiChannelImageMessage],
"bmp", comm.SingleChannelImageMessage)
self.mask = cv2.imread(self.args.mask,0)
def worker_loop(self, worker_num):
self.worker_num = worker_num
with open(self.args.classifier,"r") as fh:
print "unpickling"
self.classifier = cPickle.load(fh)
print "done"
ParallelMapper.worker_loop(self,worker_num)
def add_extra_arguments(self, parser):
ParallelMapper.add_extra_arguments(self, parser)
parser.add_argument('classifier', type=str)
parser.add_argument('--mask',
default=join(comm.DATA_ROOT, "once/roi_mask.png"))
return parser
def __init__(self):
ParallelMapper.__init__(self, ["bmp"], [comm.MultiChannelImageMessage],
"bmp", comm.SingleChannelImageMessage)
self.mask = cv2.imread(self.args.mask, 0)
def add_extra_arguments(self, parser):
ParallelMapper.add_extra_arguments(self,parser)
parser.add_argument('classifier',type=str)
parser.add_argument('--mask',default=join(comm.DATA_ROOT,"once/roi_mask.png"))
return parser
