remove numpy and jsos dependencies, and download pytqt for windows
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user