app.py

#!/usr/bin/env python

from Tkinter import *
import tkSimpleDialog,tkMessageBox
from PIL import Image, ImageTk, ImageDraw
from utils import ProcessImages

class APP(object):
    def __init__(self):
        self.top = Tk()
        self.top.title('PersonLabeling')
        
        #  self.image = Image.open('test.jpg')
        self.processimages = ProcessImages.ProcessImages('images')
        self.images = self.processimages.GetImageList()
        #  self.images_info = dict().fromkeys(self.images,[])
        self.images_info = dict()
        self.curImageIdx = 0

        self.label = Label(self.top)
        self.show_image(self.curImageIdx)

        self.label.bind('<Button-1>', self.image_click)
        self.label.bind('<ButtonRelease-1>',self.image_click_release)
        self.label.pack()

        self.prev_btn = Button(self.top, text = '(P)rev')
        self.next_btn = Button(self.top, text = '(N)ext')
        self.clean_btn = Button(self.top, text = 'Clean')
        self.save_btn = Button(self.top, text = 'Save')
        self.prev_btn.pack(side=LEFT)
        self.next_btn.pack(side=RIGHT)
        self.clean_btn.pack()
        self.save_btn.pack()
        self.prev_btn.bind('<Button-1>',self.prev_click)
        self.next_btn.bind('<Button-1>',self.next_click)
        self.clean_btn.bind('<Button-1>',self.clean_click)
        self.save_btn.bind('<Button-1>',self.save_click)
        
        self.top.bind_all('<p>', self.prev_click)
        self.top.bind_all('<n>', self.next_click)


    def show_dialog(self):
        info = tkSimpleDialog.askstring('input age and gender','split by space:\n 13 0 means age 13 male \n 20 1 means age 20 female')
        return info

    def show_image(self,image_idx):
        self.image_widget = ImageTk.PhotoImage(Image.open(self.images[image_idx]))
        self.label.configure(image = self.image_widget)

    def image_click(self, event):
        self.start_x = event.x
        self.start_y = event.y

    def image_click_release(self, event):
        info = self.show_dialog()
        if info:
            try:
                age,gender = info.split()
                int(age),int(gender)
            except:
                tkMessageBox.showwarning('error','please input valid info')
            else:
                self.save_image_info(self.curImageIdx,self.start_x,self.start_y,event.x,event.y,age,gender)
                self.draw_image(self.curImageIdx)
    
    def draw_image(self,imgIdx): 
        if self.images_info.get(self.images[imgIdx]) is None:
            return
        drawnImage = Image.open(self.images[imgIdx])
        draw = ImageDraw.Draw(drawnImage)
        for x1,y1,x2,y2,age,gender in self.images_info[self.images[imgIdx]]:
            draw.rectangle([(x1,y1),(x2,y2)], outline = (255,0,0))
            draw.text([(x2,y1)],text = str(age))
            draw.text([(x2 + 20,y1)],text = 'M' if gender == '0' else 'F')
        del draw
        self.image_widget = ImageTk.PhotoImage(drawnImage)
        self.label.configure(image = self.image_widget)
        del drawnImage

    def save_image_info(self, imageIdx, x1,y1,x2,y2,age,gender):
        if self.images_info.get(self.images[imageIdx]) is None:
            self.images_info[self.images[imageIdx]] = []
        self.images_info[self.images[imageIdx]].append((x1,y1,x2,y2,age,gender))

    def load_image_info(self, imageIdx):
        pass

    def prev_click(self, event):
        self.curImageIdx -= 1
        if self.curImageIdx < 0:
            self.curImageIdx += len(self.images)
        self.show_image(self.curImageIdx)
        self.draw_image(self.curImageIdx)

    def next_click(self, event):
        self.curImageIdx += 1
        if self.curImageIdx >= len(self.images):
            self.curImageIdx -= len(self.images)
        self.show_image(self.curImageIdx)
        self.draw_image(self.curImageIdx)

    def clean_click(self,event):
        self.images_info.pop(self.images[self.curImageIdx],None)
        self.show_image(self.curImageIdx)
        self.draw_image(self.curImageIdx)

    def save_click(self,event):
        with open('info.txt','w') as f:
            for key in self.images_info:
                for info in self.images_info[key]:
                    x1,y1,x2,y2,age,gender = info
                    print >>f,key,x1,y1,x2,y2,age,gender
def main():
    obj = APP()
    mainloop()

if __name__ == '__main__':
    main()