def write_xml(self):
imgFolderName = ustr(self.image_path).split('/')[-2]
print imgFolderName
#imgFileName = str(self.image_path).split('/')[-1]
imgFileNameWithoutExt = ustr(self.image_path).split('.')[0]
print imgFileNameWithoutExt
image = QtGui.QImage()
image.load(ustr(self.image_path))
imageShape = [
image.height(),
image.width(), 1 if image.isGrayscale() else 3
]
writer = PascalVocWriter(imgFolderName,
imgFileNameWithoutExt,
imageShape,
localImgPath=ustr(self.image_path))
#writer.verified = self.verified
item = {}
item['name'] = ''
item['flood1name'] = imgFileNameWithoutExt + '_index.bmp'
item['flood2name'] = imgFileNameWithoutExt + '_category.bmp'
writer.addObject(item['name'], item['flood1name'], item['flood2name'])
writer.save()
return
def create(self):
self.mode = self.CREATE
if self.image_path == "":
QMessageBox.information(self,"Information", u"请先打开一张图片！")
return
if os.path.exists(ustr(self.image_path)+'.xml'):
os.rm(ustr(self.image_path)+'.xml')
self.points = []
self.pre_pos = None
#1:bgr 0:grayscale
self.image = cv2.imdecode(np.fromfile(ustr(self.image_path), dtype=np.uint8), 1)
#self.image = cv2.imdecode(self.label_Image.pixmap, 1)
self.image_gray = cv2.imdecode(np.fromfile(ustr(self.image_path), dtype=np.uint8), 0)
h, w = self.image.shape[:2]
self.mask = np.zeros((h+2, w+2), np.uint8)
self.flooded = copy.deepcopy(self.image_gray)
self.flooded[:]=0
self.flooded2 = copy.deepcopy(self.image_gray)
self.flooded2[:]=0
self.flooded2_show = copy.deepcopy(self.image)
self.flooded2_show[:]=0
self.mask[:] = 0
cv2.namedWindow('image', cv2.WINDOW_KEEPRATIO | cv2.WINDOW_NORMAL)
cv2.imshow('image',self.image)
cv2.setMouseCallback('image',self.draw)
def open_previmg(self):
image_list = []
dir_path1 = ustr(self.image_path).split('/')[:-1]
current_image = ustr(self.image_path).split('/')[-1]
dir_path = '/'.join(dir_path1)
dir_items = os.listdir(dir_path)
for item in dir_items:
if item.endswith('.jpg') or item.endswith('.png') or item.endswith(
'.jpeg'):
image_list.append(item)
#print image_list
#print image_list.index(current_image)
if image_list.index(current_image) == 0:
QtGui.QMessageBox.information(self, "Information", u"已经是第一张了！")
return
current_image_index = image_list.index(current_image) - 1
next_image = dir_path + '/' + image_list[current_image_index]
png = QtGui.QPixmap(next_image).scaled(self.label_Image.width(),
self.label_Image.height())
self.label_Image.setPixmap(png)
self.image_path = next_image
#self.label_imagepath.setText(self.image_path)
self.setWindowTitle(
_translate("MainWindow", "LableMe" + " " + ustr(self.image_path),
None))
def saveFile(self, _value=False):
if self.defaultSaveDir is not None and len(ustr(self.defaultSaveDir)):
# print('handle the image:' + self.filePath)
if self.filePath:
imgFileName = os.path.basename(self.filePath)
savedFileName = os.path.splitext(imgFileName)[0] + LabelFile.suffix
savedPath = os.path.join(ustr(self.defaultSaveDir), savedFileName)
self._saveFile(savedPath)
else:
self._saveFile(self.filePath if self.labelFile
else self.saveFileDialog())
def open(self):
img_name = ustr(
QtGui.QFileDialog.getOpenFileName(
self, 'Open file', ".",
"*.jpg;;*.png;;*.jpeg;;*.bmp;;All Files (*)"))
#print unicode(img_name, 'ascii')
print(img_name)
self.png = QtGui.QPixmap(img_name).scaled(self.label_Image.width(),
self.label_Image.height())
self.label_Image.setPixmap(self.png)
self.image_path = img_name
#self.image_path.setText(self.image_path)
self.setWindowTitle(
_translate("MainWindow", "LabelMe" + " " + ustr(self.image_path),
None))
def saveLabels(self, annotationFilePath):
annotationFilePath = ustr(annotationFilePath)
if self.labelFile is None:
self.labelFile = LabelFile()
self.labelFile.verified = self.canvas.verified
def format_shape(s):
return dict(label=s.label,
line_color=s.line_color.getRgb()
if s.line_color != self.lineColor else None,
fill_color=s.fill_color.getRgb()
if s.fill_color != self.fillColor else None,
points=[(p.x(), p.y()) for p in s.points],
# add chris
difficult = s.difficult)
shapes = [format_shape(shape) for shape in self.canvas.shapes]
# Can add differrent annotation formats here
try:
if self.usingPascalVocFormat is True:
print ('Img: ' + self.filePath +
' -> Its xml: ' + annotationFilePath)
self.labelFile.savePascalVocFormat(annotationFilePath, shapes, self.filePath, self.imageData,
self.lineColor.getRgb(), self.fillColor.getRgb())
else:
self.labelFile.save(annotationFilePath, shapes, self.filePath, self.imageData,
self.lineColor.getRgb(), self.fillColor.getRgb())
return True
except LabelFileError as e:
self.errorMessage(u'Error saving label data',
u'<b>%s</b>' % e)
return False
def openDir(self, _value=False):
if not self.mayContinue():
return
path = os.path.dirname(self.filePath)\
if self.filePath else '.'
if self.lastOpenDir is not None and len(self.lastOpenDir) > 1:
path = self.lastOpenDir
dirpath = ustr(QFileDialog.getExistingDirectory(self,
'%s - Open Directory' % __appname__, path, QFileDialog.ShowDirsOnly
| QFileDialog.DontResolveSymlinks))
if dirpath is not None and len(dirpath) > 1:
self.lastOpenDir = dirpath
self.dirname = dirpath
self.filePath = None
self.fileListWidget.clear()
self.mImgList = self.scanAllImages(dirpath)
self.openNextImg()
for imgPath in self.mImgList:
item = QListWidgetItem(imgPath)
self.fileListWidget.addItem(item)
def changeSavedir(self, _value=False):
if self.defaultSaveDir is not None:
path = ustr(self.defaultSaveDir)
else:
path = '.'
dirpath = ustr(QFileDialog.getExistingDirectory(self,
'%s - Save to the directory' % __appname__, path, QFileDialog.ShowDirsOnly
| QFileDialog.DontResolveSymlinks))
if dirpath is not None and len(dirpath) > 1:
self.defaultSaveDir = dirpath
self.statusBar().showMessage('%s . Annotation will be saved to %s' %
('Change saved folder', self.defaultSaveDir))
self.statusBar().show()
def appendObjects(self, top):
for each_object in self.boxlist:
object_item = SubElement(top, 'object')
name = SubElement(object_item, 'name')
name.text = ustr(each_object['name'])
pose = SubElement(object_item, 'pose')
pose.text = "Unspecified"
truncated = SubElement(object_item, 'truncated')
if int(float(each_object['ymax'])) == int(float(self.imgSize[0])) or (int(float(each_object['ymin'])) == 1):
truncated.text = "1" # max == height or min
elif (int(float(each_object['xmax'])) == int(float(self.imgSize[1]))) or (int(float(each_object['xmin'])) == 1):
truncated.text = "1" # max == width or min
else:
truncated.text = "0"
difficult = SubElement(object_item, 'difficult')
difficult.text = str(bool(each_object['difficult']) & 1)
bndbox = SubElement(object_item, 'bndbox')
xmin = SubElement(bndbox, 'xmin')
xmin.text = str(each_object['xmin'])
ymin = SubElement(bndbox, 'ymin')
ymin.text = str(each_object['ymin'])
xmax = SubElement(bndbox, 'xmax')
xmax.text = str(each_object['xmax'])
ymax = SubElement(bndbox, 'ymax')
ymax.text = str(each_object['ymax'])
def generateColorByText(text):
s = str(ustr(text))
hashCode = int(hashlib.sha256(s.encode('utf-8')).hexdigest(), 16)
r = int((hashCode / 255) % 255)
g = int((hashCode / 65025) % 255)
b = int((hashCode / 16581375) % 255)
return QColor(r, g, b, 100)
def closeEvent(self, event):
if not self.mayContinue():
event.ignore()
s = self.settings
# If it loads images from dir, don't load it at the begining
if self.dirname is None:
s['filename'] = self.filePath if self.filePath else ''
else:
s['filename'] = ''
s['window/size'] = self.size()
s['window/position'] = self.pos()
s['window/state'] = self.saveState()
s['line/color'] = self.lineColor
s['fill/color'] = self.fillColor
s['recentFiles'] = self.recentFiles
s['advanced'] = not self._beginner
if self.defaultSaveDir is not None and len(self.defaultSaveDir) > 1:
s['savedir'] = ustr(self.defaultSaveDir)
else:
s['savedir'] = ""
if self.lastOpenDir is not None and len(self.lastOpenDir) > 1:
s['lastOpenDir'] = self.lastOpenDir
else:
s['lastOpenDir'] = ""
def openFile(self, _value=False):
if not self.mayContinue():
return
path = os.path.dirname(ustr(self.filePath)) if self.filePath else '.'
formats = ['*.%s' % fmt.data().decode("ascii").lower() for fmt in QImageReader.supportedImageFormats()]
filters = "Image & Label files (%s)" % ' '.join(formats + ['*%s' % LabelFile.suffix])
filename = QFileDialog.getOpenFileName(self, '%s - Choose Image or Label file' % __appname__, path, filters)
if filename:
if isinstance(filename, (tuple, list)):
filename = filename[0]
self.loadFile(filename)
def scanAllImages(self, folderPath):
extensions = ['.jpeg', '.jpg', '.png', '.bmp']
images = []
for root, dirs, files in os.walk(folderPath):
for file in files:
if file.lower().endswith(tuple(extensions)):
relatviePath = os.path.join(root, file)
path = ustr(os.path.abspath(relatviePath))
images.append(path)
images.sort(key=lambda x: x.lower())
return images
def delete_line(self):
if self.pre_pos is not None:
point1 = self.points.pop()
#print point1
point2 = self.points.pop()
#print point2
cv2.circle(self.image,point1,2,(0,0,0),2)
cv2.circle(self.mask,point1,2,(0,0,0),2)
cv2.circle(self.image,point2,2,(0,0,0),2)
cv2.circle(self.mask,point2,2,(0,0,0),2)
cv2.line(ustr(self.image), point2, point1, (0,0,0),2)
cv2.line(self.mask, point2, point1, (0,0,0), 2)
cv2.imshow('image', self.image)
self.pre_pos = None
def openAnnotation(self, _value=False):
if self.filePath is None:
return
path = os.path.dirname(ustr(self.filePath))\
if self.filePath else '.'
if self.usingPascalVocFormat:
filters = "Open Annotation XML file (%s)" % \
' '.join(['*.xml'])
filename = QFileDialog.getOpenFileName(self,'%s - Choose a xml file' % __appname__, path, filters)
if filename:
if isinstance(filename, (tuple, list)):
filename = filename[0]
self.loadPascalXMLByFilename(filename)
def _cast(self, key, value):
# XXX: Very nasty way of converting types to QVariant methods :P
t = self.types.get(key)
if t is not None and t != QVariant:
if t is str:
return ustr(value)
else:
try:
method = getattr(QVariant, re.sub(
'^Q', 'to', t.__name__, count=1))
return method(value)
except AttributeError as e:
# print(e)
return value
return value
def render_blocks(blocks, md):
rendered = []
append=rendered.append
for section in blocks:
if type(section) is TupleType:
l=len(section)
if l==1:
# Simple var
section=section[0]
if type(section) is StringType: section=md[section]
else: section=section(md)
section=ustr(section)
else:
# if
cache={}
md._push(cache)
try:
i=0
m=l-1
while i < m:
cond=section[i]
if type(cond) is StringType:
n=cond
try:
cond=md[cond]
cache[n]=cond
except KeyError, v:
v=str(v)
if n != v: raise KeyError, v, sys.exc_traceback
cond=None
else: cond=cond(md)
if cond:
section=section[i+1]
if section: section=render_blocks(section,md)
else: section=''
m=0
break
i=i+2
if m:
if i==m: section=render_blocks(section[i],md)
else: section=''
finally: md._pop()
elif type(section) is not StringType and type(section) is not UnicodeType:
section=section(md)
def render_blocks(blocks, md):
rendered = []
append = rendered.append
for section in blocks:
if type(section) is TupleType:
l = len(section)
if l == 1:
# Simple var
section = section[0]
if type(section) is StringType: section = md[section]
else: section = section(md)
section = ustr(section)
else:
# if
cache = {}
md._push(cache)
try:
i = 0
m = l - 1
while i < m:
cond = section[i]
if type(cond) is StringType:
n = cond
try:
cond = md[cond]
cache[n] = cond
except KeyError, v:
v = str(v)
if n != v: raise KeyError, v, sys.exc_traceback
cond = None
else:
cond = cond(md)
if cond:
section = section[i + 1]
if section: section = render_blocks(section, md)
else: section = ''
m = 0
break
i = i + 2
if m:
if i == m: section = render_blocks(section[i], md)
else: section = ''
finally:
md._pop()
def draw(self,event,x,y,flags,param):
#self.save_file()
fixed_range=True
path = ustr(self.image_path).split('.')[0]
suffix = 'bmp'
#print path
mask_name = path + '_mask.' + suffix
flood_name = path + '_index.' + suffix
flood2_name = path + '_category.' + suffix
self.choose_category.emit()
if event == cv2.EVENT_LBUTTONDOWN:
self.points.append((x,y))
#cv2.circle(self.label_Image.pixmap,(x,y),2,(0,0,255),2)
cv2.circle(self.image,(x,y),2,(0,0,255),2)
cv2.circle(self.mask,(x,y),2,(255,255,255),2)
#print self.points
#print self.pre_pos
if self.pre_pos is not None:
cv2.line(ustr(self.image), self.pre_pos, (x,y), (0,255,0),2)
cv2.line(self.mask, self.pre_pos, (x,y), (255,255,255), 2)
if self.isEqual(self.points[0],self.points[-1]):
cv2.line(ustr(self.image), (x,y), self.points[0], (0,255,0),2)
cv2.line(self.mask, (x,y), self.points[0], (255,255,255), 2)
self.pre_pos = None
self.points = []
else:
self.pre_pos = (x, y)
else:
#self.first_pos = (x, y)
self.pre_pos = (x, y)
self.mask_ = self.mask.copy()
self.mask2_ = self.mask.copy()
cv2.imshow('image', self.image)
#cv2.imshow('mask', self.mask)
#self.pre_pos = (x, y)
#elif event == cv2.EVENT_LBUTTONDBLCLK:
# QtGui.QMessageBox.information(self,"Information", u"确定删除？")
elif event == cv2.EVENT_RBUTTONUP:
self.pre_pos = None
#print(self.mask.dtype)
fixed_range=True
flags=8
#print 'step 1'
#点击到区域外撤销
tmp_mask2_ = self.mask.copy()
tmp = self.bgr_list[ustr(self.object)]
if self.flooded.item(y,x) != 0 or self.flooded2.item(y,x) != 0 :
cv2.floodFill(self.flooded2_show, tmp_mask2_, (x,y), (0,0,0), (0,)*3, (0,)*3, flags)
cv2.floodFill(self.flooded, self.mask_, (x,y), 0, (255,)*3, (160,)*3, flags)
cv2.floodFill(self.flooded2, self.mask2_, (x,y), 0, (255,)*3, (160,)*3, flags)
#cv2.imshow('index',self.flooded)
cv2.imshow('category',self.flooded2_show)
cv2.imencode('.bmp',self.flooded)[1].tofile(ustr(flood_name))
cv2.imencode('.bmp',self.flooded2)[1].tofile(ustr(flood2_name))
#self.index_num = self.index_num - 1
else:
self.index_num = self.index_num + 1
tmp_mask = self.mask.copy()
print self.index_num
#cv2.floodFill(self.flooded, tmp_mask, (x,y), (random.randint(0,255), random.randint(0,255), random.randint(0,255)), (0,)*3, (0,)*3, 4)
cv2.floodFill(self.flooded, tmp_mask, (x,y), self.index_num, (0,)*3, (0,)*3, 4)
mask_gray=self.mask.copy()
res,thres=cv2.threshold(mask_gray,0,255,cv2.THRESH_BINARY)
thresgray=cv2.cvtColor(thres,cv2.COLOR_GRAY2BGR)
#cv2.imshow('index',self.flooded)
cv2.imencode('.bmp',self.flooded)[1].tofile(ustr(flood_name))
tmp_mask2 = self.mask.copy()
tmp_mask3 = self.mask.copy()
print self.bgr_list[unicode(self.object)]
tmp = self.bgr_list[unicode(self.object)][1:-1].split(',')
b = int(tmp[0])
g = int(tmp[1])
r = int(tmp[2])
print self.gray_list[unicode(self.object)]
cv2.floodFill(self.flooded2_show, tmp_mask3, (x,y), (b, g, r), (0,)*3, (0,)*3, flags)
cv2.floodFill(self.flooded2, tmp_mask2, (x,y), int(self.gray_list[unicode(self.object)]), (0,)*3, (0,)*3, flags)
cv2.imshow('category',self.flooded2_show)
cv2.imencode('.bmp',self.flooded2)[1].tofile(ustr(flood2_name))
def html_quote(v, name='(Unknown name)', md={}):
return escape(ustr(v), 1)
def save_file(self):
if self.image_path:
image_file_name = ustr(self.image_path).split('.')[0]
saved_file_name = image_file_name + '.xml'
def fileitemDoubleClicked(self, item=None):
currIndex = self.mImgList.index(ustr(item.text()))
if currIndex < len(self.mImgList):
filename = self.mImgList[currIndex]
if filename:
self.loadFile(filename)
def loadFile(self, filePath=None):
"""Load the specified file, or the last opened file if None."""
self.resetState()
self.canvas.setEnabled(False)
if filePath is None:
filePath = self.settings.get('filename')
unicodeFilePath = ustr(filePath)
# Tzutalin 20160906 : Add file list and dock to move faster
# Highlight the file item
if unicodeFilePath and self.fileListWidget.count() > 0:
index = self.mImgList.index(unicodeFilePath)
fileWidgetItem = self.fileListWidget.item(index)
fileWidgetItem.setSelected(True)
if unicodeFilePath and os.path.exists(unicodeFilePath):
if LabelFile.isLabelFile(unicodeFilePath):
try:
self.labelFile = LabelFile(unicodeFilePath)
except LabelFileError as e:
self.errorMessage(u'Error opening file',
(u"<p><b>%s</b></p>"
u"<p>Make sure <i>%s</i> is a valid label file.")
% (e, unicodeFilePath))
self.status("Error reading %s" % unicodeFilePath)
return False
self.imageData = self.labelFile.imageData
self.lineColor = QColor(*self.labelFile.lineColor)
self.fillColor = QColor(*self.labelFile.fillColor)
else:
# Load image:
# read data first and store for saving into label file.
self.imageData = read(unicodeFilePath, None)
self.labelFile = None
image = QImage.fromData(self.imageData)
if image.isNull():
self.errorMessage(u'Error opening file',
u"<p>Make sure <i>%s</i> is a valid image file." % unicodeFilePath)
self.status("Error reading %s" % unicodeFilePath)
return False
self.status("Loaded %s" % os.path.basename(unicodeFilePath))
self.image = image
self.filePath = unicodeFilePath
self.canvas.loadPixmap(QPixmap.fromImage(image))
if self.labelFile:
self.loadLabels(self.labelFile.shapes)
self.setClean()
self.canvas.setEnabled(True)
self.adjustScale(initial=True)
self.paintCanvas()
self.addRecentFile(self.filePath)
self.toggleActions(True)
# Label xml file and show bound box according to its filename
if self.usingPascalVocFormat is True:
if self.defaultSaveDir is not None:
basename = os.path.basename(
os.path.splitext(self.filePath)[0]) + XML_EXT
xmlPath = os.path.join(self.defaultSaveDir, basename)
self.loadPascalXMLByFilename(xmlPath)
else:
xmlPath = filePath.split(".")[0] + XML_EXT
if os.path.isfile(xmlPath):
self.loadPascalXMLByFilename(xmlPath)
self.setWindowTitle(__appname__ + ' ' + filePath)
# Default : select last item if there is at least one item
if self.labelList.count():
self.labelList.setCurrentItem(self.labelList.item(self.labelList.count()-1))
self.labelList.setItemSelected(self.labelList.item(self.labelList.count()-1), True)
self.canvas.setFocus(True)
return True
return False
def html_quote(v, name='(Unknown name)', md={}):
return escape(ustr(v), 1)
def __init__(self, defaultFilename=None):
super(MainWindow, self).__init__()
self.setWindowTitle(__appname__)
# Save as Pascal voc xml
self.defaultSaveDir = None
self.usingPascalVocFormat = True
if self.usingPascalVocFormat:
LabelFile.suffix = '.xml'
# For loading all image under a directory
self.mImgList = []
self.dirname = None
self.labelHist = []
self.lastOpenDir = None
# Whether we need to save or not.
self.dirty = False
# Enble auto saving if pressing next
self.autoSaving = True
self._noSelectionSlot = False
self._beginner = True
self.screencastViewer = "firefox"
self.screencast = "https://youtu.be/p0nR2YsCY_U"
self.loadPredefinedClasses()
# Main widgets and related state.
self.labelDialog = LabelDialog(parent=self, listItem=self.labelHist)
self.labelList = QListWidget()
self.itemsToShapes = {}
self.shapesToItems = {}
self.prevLabelText = ''
self.labelList.itemActivated.connect(self.labelSelectionChanged)
self.labelList.itemSelectionChanged.connect(self.labelSelectionChanged)
self.labelList.itemDoubleClicked.connect(self.editLabel)
# Connect to itemChanged to detect checkbox changes.
self.labelList.itemChanged.connect(self.labelItemChanged)
listLayout = QVBoxLayout()
listLayout.setContentsMargins(0, 0, 0, 0)
listLayout.addWidget(self.labelList)
self.editButton = QToolButton()
self.editButton.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
# Add chris
self.diffcButton = QCheckBox("Difficult")
self.diffcButton.setChecked(False)
self.diffcButton.stateChanged.connect(self.btnstate)
self.labelListContainer = QWidget()
self.labelListContainer.setLayout(listLayout)
listLayout.addWidget(self.editButton) # , 0, Qt.AlignCenter)
# Add chris
listLayout.addWidget(self.diffcButton)
listLayout.addWidget(self.labelList)
self.dock = QDockWidget(u'Box Labels', self)
self.dock.setObjectName(u'Labels')
self.dock.setWidget(self.labelListContainer)
# Tzutalin 20160906 : Add file list and dock to move faster
self.fileListWidget = QListWidget()
self.fileListWidget.itemDoubleClicked.connect(
self.fileitemDoubleClicked)
filelistLayout = QVBoxLayout()
filelistLayout.setContentsMargins(0, 0, 0, 0)
filelistLayout.addWidget(self.fileListWidget)
self.fileListContainer = QWidget()
self.fileListContainer.setLayout(filelistLayout)
self.filedock = QDockWidget(u'File List', self)
self.filedock.setObjectName(u'Files')
self.filedock.setWidget(self.fileListContainer)
self.zoomWidget = ZoomWidget()
self.colorDialog = ColorDialog(parent=self)
self.canvas = Canvas()
self.canvas.zoomRequest.connect(self.zoomRequest)
scroll = QScrollArea()
scroll.setWidget(self.canvas)
scroll.setWidgetResizable(True)
self.scrollBars = {
Qt.Vertical: scroll.verticalScrollBar(),
Qt.Horizontal: scroll.horizontalScrollBar()
}
self.canvas.scrollRequest.connect(self.scrollRequest)
self.canvas.newShape.connect(self.newShape)
self.canvas.shapeMoved.connect(self.setDirty)
self.canvas.selectionChanged.connect(self.shapeSelectionChanged)
self.canvas.drawingPolygon.connect(self.toggleDrawingSensitive)
self.setCentralWidget(scroll)
self.addDockWidget(Qt.RightDockWidgetArea, self.dock)
# Tzutalin 20160906 : Add file list and dock to move faster
self.addDockWidget(Qt.RightDockWidgetArea, self.filedock)
self.dockFeatures = QDockWidget.DockWidgetClosable\
| QDockWidget.DockWidgetFloatable
self.dock.setFeatures(self.dock.features() ^ self.dockFeatures)
# Actions
action = partial(newAction, self)
quit = action('&Quit', self.close,
'Ctrl+Q', 'quit', u'Quit application')
open = action('&Open', self.openFile,
'Ctrl+O', 'open', u'Open image or label file')
opendir = action('&Open Dir', self.openDir,
'Ctrl+u', 'open', u'Open Dir')
changeSavedir = action('&Change default saved Annotation dir', self.changeSavedir,
'Ctrl+r', 'open', u'Change default saved Annotation dir')
openAnnotation = action('&Open Annotation', self.openAnnotation,
'Ctrl+Shift+O', 'openAnnotation', u'Open Annotation')
openNextImg = action('&Next Image', self.openNextImg,
'd', 'next', u'Open Next')
openPrevImg = action('&Prev Image', self.openPrevImg,
'a', 'prev', u'Open Prev')
verify = action('&Verify Image', self.verifyImg,
'space', 'verify', u'Verify Image')
save = action('&Save', self.saveFile,
'Ctrl+S', 'save', u'Save labels to file', enabled=False)
saveAs = action('&Save As', self.saveFileAs,
'Ctrl+Shift+S', 'save-as', u'Save labels to a different file',
enabled=False)
close = action('&Close', self.closeFile,
'Ctrl+W', 'close', u'Close current file')
color1 = action('Box &Line Color', self.chooseColor1,
'Ctrl+L', 'color_line', u'Choose Box line color')
color2 = action('Box &Fill Color', self.chooseColor2,
'Ctrl+Shift+L', 'color', u'Choose Box fill color')
createMode = action('Create\nRectBox', self.setCreateMode,
'Ctrl+N', 'new', u'Start drawing Boxs', enabled=False)
editMode = action('&Edit\nRectBox', self.setEditMode,
'Ctrl+J', 'edit', u'Move and edit Boxs', enabled=False)
create = action('Create\nRectBox', self.createShape,
'w', 'new', u'Draw a new Box', enabled=False)
delete = action('Delete\nRectBox', self.deleteSelectedShape,
'Delete', 'delete', u'Delete', enabled=False)
copy = action('&Duplicate\nRectBox', self.copySelectedShape,
'Ctrl+D', 'copy', u'Create a duplicate of the selected Box',
enabled=False)
advancedMode = action('&Advanced Mode', self.toggleAdvancedMode,
'Ctrl+Shift+A', 'expert', u'Switch to advanced mode',
checkable=True)
hideAll = action('&Hide\nRectBox', partial(self.togglePolygons, False),
'Ctrl+H', 'hide', u'Hide all Boxs',
enabled=False)
showAll = action('&Show\nRectBox', partial(self.togglePolygons, True),
'Ctrl+A', 'hide', u'Show all Boxs',
enabled=False)
help = action('&Tutorial', self.tutorial, 'Ctrl+T', 'help',
u'Show demos')
zoom = QWidgetAction(self)
zoom.setDefaultWidget(self.zoomWidget)
self.zoomWidget.setWhatsThis(
u"Zoom in or out of the image. Also accessible with"
" %s and %s from the canvas." % (fmtShortcut("Ctrl+[-+]"),
fmtShortcut("Ctrl+Wheel")))
self.zoomWidget.setEnabled(False)
zoomIn = action('Zoom &In', partial(self.addZoom, 10),
'Ctrl++', 'zoom-in', u'Increase zoom level', enabled=False)
zoomOut = action('&Zoom Out', partial(self.addZoom, -10),
'Ctrl+-', 'zoom-out', u'Decrease zoom level', enabled=False)
zoomOrg = action('&Original size', partial(self.setZoom, 100),
'Ctrl+=', 'zoom', u'Zoom to original size', enabled=False)
fitWindow = action('&Fit Window', self.setFitWindow,
'Ctrl+F', 'fit-window', u'Zoom follows window size',
checkable=True, enabled=False)
fitWidth = action('Fit &Width', self.setFitWidth,
'Ctrl+Shift+F', 'fit-width', u'Zoom follows window width',
checkable=True, enabled=False)
# Group zoom controls into a list for easier toggling.
zoomActions = (self.zoomWidget, zoomIn, zoomOut,
zoomOrg, fitWindow, fitWidth)
self.zoomMode = self.MANUAL_ZOOM
self.scalers = {
self.FIT_WINDOW: self.scaleFitWindow,
self.FIT_WIDTH: self.scaleFitWidth,
# Set to one to scale to 100% when loading files.
self.MANUAL_ZOOM: lambda: 1,
}
edit = action('&Edit Label', self.editLabel,
'Ctrl+E', 'edit', u'Modify the label of the selected Box',
enabled=False)
self.editButton.setDefaultAction(edit)
shapeLineColor = action('Shape &Line Color', self.chshapeLineColor,
icon='color_line', tip=u'Change the line color for this specific shape',
enabled=False)
shapeFillColor = action('Shape &Fill Color', self.chshapeFillColor,
icon='color', tip=u'Change the fill color for this specific shape',
enabled=False)
labels = self.dock.toggleViewAction()
labels.setText('Show/Hide Label Panel')
labels.setShortcut('Ctrl+Shift+L')
# Lavel list context menu.
labelMenu = QMenu()
addActions(labelMenu, (edit, delete))
self.labelList.setContextMenuPolicy(Qt.CustomContextMenu)
self.labelList.customContextMenuRequested.connect(
self.popLabelListMenu)
# Store actions for further handling.
self.actions = struct(save=save, saveAs=saveAs, open=open, close=close,
lineColor=color1, fillColor=color2,
create=create, delete=delete, edit=edit, copy=copy,
createMode=createMode, editMode=editMode, advancedMode=advancedMode,
shapeLineColor=shapeLineColor, shapeFillColor=shapeFillColor,
zoom=zoom, zoomIn=zoomIn, zoomOut=zoomOut, zoomOrg=zoomOrg,
fitWindow=fitWindow, fitWidth=fitWidth,
zoomActions=zoomActions,
fileMenuActions=(
open, opendir, save, saveAs, close, quit),
beginner=(), advanced=(),
editMenu=(edit, copy, delete,
None, color1, color2),
beginnerContext=(create, edit, copy, delete),
advancedContext=(createMode, editMode, edit, copy,
delete, shapeLineColor, shapeFillColor),
onLoadActive=(
close, create, createMode, editMode),
onShapesPresent=(saveAs, hideAll, showAll))
self.menus = struct(
file=self.menu('&File'),
edit=self.menu('&Edit'),
view=self.menu('&View'),
help=self.menu('&Help'),
recentFiles=QMenu('Open &Recent'),
labelList=labelMenu)
addActions(self.menus.file,
(open, opendir, changeSavedir, openAnnotation, self.menus.recentFiles, save, saveAs, close, None, quit))
addActions(self.menus.help, (help,))
addActions(self.menus.view, (
labels, advancedMode, None,
hideAll, showAll, None,
zoomIn, zoomOut, zoomOrg, None,
fitWindow, fitWidth))
self.menus.file.aboutToShow.connect(self.updateFileMenu)
# Custom context menu for the canvas widget:
addActions(self.canvas.menus[0], self.actions.beginnerContext)
addActions(self.canvas.menus[1], (
action('&Copy here', self.copyShape),
action('&Move here', self.moveShape)))
self.tools = self.toolbar('Tools')
self.actions.beginner = (
open, opendir, openNextImg, openPrevImg, verify, save, None, create, copy, delete, None,
zoomIn, zoom, zoomOut, fitWindow, fitWidth)
self.actions.advanced = (
open, save, None,
createMode, editMode, None,
hideAll, showAll)
self.statusBar().showMessage('%s started.' % __appname__)
self.statusBar().show()
# Application state.
self.image = QImage()
self.filePath = ustr(defaultFilename)
self.recentFiles = []
self.maxRecent = 7
self.lineColor = None
self.fillColor = None
self.zoom_level = 100
self.fit_window = False
# Add Chris
self.difficult = False
# XXX: Could be completely declarative.
# Restore application settings.
if have_qstring():
types = {
'filename': QString,
'recentFiles': QStringList,
'window/size': QSize,
'window/position': QPoint,
'window/geometry': QByteArray,
'line/color': QColor,
'fill/color': QColor,
'advanced': bool,
# Docks and toolbars:
'window/state': QByteArray,
'savedir': QString,
'lastOpenDir': QString,
}
else:
types = {
'filename': str,
'recentFiles': list,
'window/size': QSize,
'window/position': QPoint,
'window/geometry': QByteArray,
'line/color': QColor,
'fill/color': QColor,
'advanced': bool,
# Docks and toolbars:
'window/state': QByteArray,
'savedir': str,
'lastOpenDir': str,
}
self.settings = settings = Settings(types)
self.recentFiles = list(settings.get('recentFiles', []))
size = settings.get('window/size', QSize(600, 500))
position = settings.get('window/position', QPoint(0, 0))
self.resize(size)
self.move(position)
saveDir = ustr(settings.get('savedir', None))
self.lastOpenDir = ustr(settings.get('lastOpenDir', None))
if saveDir is not None and os.path.exists(saveDir):
self.defaultSaveDir = saveDir
self.statusBar().showMessage('%s started. Annotation will be saved to %s' %
(__appname__, self.defaultSaveDir))
self.statusBar().show()
# or simply:
# self.restoreGeometry(settings['window/geometry']
self.restoreState(settings.get('window/state', QByteArray()))
self.lineColor = QColor(settings.get('line/color', Shape.line_color))
self.fillColor = QColor(settings.get('fill/color', Shape.fill_color))
Shape.line_color = self.lineColor
Shape.fill_color = self.fillColor
# Add chris
Shape.Difficult = self.difficult
def xbool(x):
if isinstance(x, QVariant):
return x.toBool()
return bool(x)
if xbool(settings.get('advanced', False)):
self.actions.advancedMode.setChecked(True)
self.toggleAdvancedMode()
# Populate the File menu dynamically.
self.updateFileMenu()
# Since loading the file may take some time, make sure it runs in the
# background.
self.queueEvent(partial(self.loadFile, self.filePath or ""))
# Callbacks:
self.zoomWidget.valueChanged.connect(self.paintCanvas)
self.populateModeActions()
