remove numpy and jsos dependencies, and download pytqt for windows

This commit is contained in:
tzutalin
2016-12-07 14:39:45 +08:00
parent 16e6742094
commit 3bbf5e23e5
3 changed files with 11 additions and 34 deletions
-34
View File
@@ -4,8 +4,6 @@
from PyQt4.QtGui import QImage
from base64 import b64encode, b64decode
from pascal_voc_io import PascalVocWriter
import json
import numpy
import os.path
import sys
@@ -23,38 +21,6 @@ class LabelFile(object):
if filename is not None:
self.load(filename)
def load(self, filename):
try:
with open(filename, 'rb') as f:
data = json.load(f)
imagePath = data['imagePath']
imageData = b64decode(data['imageData'])
lineColor = data['lineColor']
fillColor = data['fillColor']
shapes = ((s['label'], s['points'], s['line_color'], s['fill_color'])\
for s in data['shapes'])
# Only replace data after everything is loaded.
self.shapes = shapes
self.imagePath = imagePath
self.imageData = imageData
self.lineColor = lineColor
self.fillColor = fillColor
except Exception, e:
raise LabelFileError(e)
def save(self, filename, shapes, imagePath, imageData,
lineColor=None, fillColor=None):
try:
with open(filename, 'wb') as f:
json.dump(dict(
shapes=shapes,
lineColor=lineColor, fillColor=fillColor,
imagePath=imagePath,
imageData=b64encode(imageData)),
f, ensure_ascii=True, indent=2)
except Exception, e:
raise LabelFileError(e)
def savePascalVocFormat(self, filename, shapes, imagePath, imageData,
lineColor=None, fillColor=None, databaseSrc=None):
imgFolderPath = os.path.dirname(imagePath)