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