python图像标注工具_深度学习python图像标记工具labelTool

# -*- coding:utf-8 -*-

#-------------------------------------------------------------------------------

# Name:        Object bounding box label tool

# Purpose:     Label object bboxes for ImageNet Detection data

# Author:      Qiushi

# Created:     06/06/2014

#

#-------------------------------------------------------------------------------

from __future__ import division

from Tkinter import *

import tkMessageBox

from PIL import Image, ImageTk

import os

import glob

import random

w0 = 1; #图片原始宽度

h0 = 1; #图片原始高度

# colors for the bboxes

COLORS = ['red', 'blue', 'yellow', 'pink', 'cyan', 'green', 'black']

# image sizes for the examples

SIZE = 256, 256

#指定缩放后的图像大小

DEST_SIZE = 500, 500

class LabelTool():

def __init__(self, master):

# set up the main frame

self.parent = master

self.parent.title("LabelTool")

self.frame = Frame(self.parent)

self.frame.pack(fill=BOTH, expand=1)

self.parent.resizable(width = TRUE, height = TRUE)

# initialize global state

self.imageDir = ''

self.imageList= []

self.egDir = ''

self.egList = []

self.outDir = ''

self.cur = 0

self.total = 0

self.category = 0

self.imagename = ''

self.labelfilename = ''

self.tkimg = None

# initialize mouse state

self.STATE = {}

self.STATE['click'] = 0

self.STATE['x'], self.STATE['y'] = 0, 0

# reference to bbox

self.bboxIdList = []

self.bboxId = None

self.bboxList = []

self.hl = None

self.vl = None

# ----------------- GUI stuff ---------------------

# dir entry & load

self.label = Label(self.frame, text = "Image Dir:")

self.label.grid(row = 0, column = 0, sticky = E)

self.entry = Entry(self.frame)

self.entry.grid(row = 0, column = 1, sticky = W+E)

self.ldBtn = Button(self.frame, text = "Load", command = self.loadDir)

self.ldBtn.grid(row = 0, column = 2, sticky = W+E)

# main panel for labeling

self.mainPanel = Canvas(self.frame, cursor='tcross')

self.mainPanel.bind("", self.mouseClick)

self.mainPanel.bind("", self.mouseMove)

self.parent.bind("", self.cancelBBox)  # press  to cancel current bbox

self.parent.bind("s", self.cancelBBox)

self.parent.bind("a", self.prevImage) # press 'a' to go backforward

self.parent