GenderThread.py 文件源码

def run(self):
        caffe.set_mode_gpu()

        while self.videoThread.isTerminated() == False:

        while self.videoThread.isTerminated() == False and self.videoThread.getEventReady() == True:
                time.sleep(0.1)
                print("Gender recognition sleep")

            #print "Detecting..."
            crop = None

            while crop == None:

                crop, rectangle = self.videoThread.getCropEx(1)
                time.sleep(0.05)
                if crop == None: # No crops available yet
                    time.sleep(0.1)

            crop = crop.astype(np.float32)

            propabilities = self.gender_net.predict([crop], oversample = False).ravel() #[Male, Female]
            self.videoThread.setGender(propabilities)