def script():
arguments = docopt(__doc__)
if arguments.get('--version') or arguments.get('-v'):
print(__version__)
elif arguments.get('m2j'):
logger.info("Masks to jsons")
print("<==== please input origin image path ====>")
inputOriimgPath = input()
print("<==== please input mask path ====>")
inputMaskPath = input()
if not os.path.exists(inputMaskPath):
raise FileNotFoundError('mask folder is not exist')
savePath = inputMaskPath
print("<==== please input yaml path (can be blank,better dont) ====>")
inputYamlPath = input()
getJsons(inputOriimgPath,inputMaskPath,savePath,inputYamlPath)
print("Done!")
elif arguments.get('m2x'):
logger.info("Masks to xmls")
print("<==== please input origin image path ====>")
inputOriimgPath = input()
print("<==== please input mask path ====>")
inputMaskPath = input()
if not os.path.exists(inputMaskPath):
raise FileNotFoundError('mask folder is not exist')
savePath = inputMaskPath
getXmls(inputOriimgPath,inputMaskPath,savePath)
print("Done!")
elif arguments.get('j2m'):
logger.info("Jsons to masks")
print("<==== please input json files path ====>")
inputJsonPath = input()
processor(inputJsonPath)
print('Done!')
elif arguments.get('j2x'):
logger.info("Json to xml (single file supported if version<=0.3)")
print("<==== please input json file path ====>")
inputJsonPath = input()
j2xConvert(inputJsonPath)
print('Done!')
elif arguments.get('aug'):
# logger.info("Image augmentation (single file supported if version<=0.3)")
if not arguments.get('--nolabel'):
print("<==== please input image file path ====>")
inputFilePath = input()
print("<==== please input json file path ====>")
inputJsonPath = input()
imgAug_withLabels(inputFilePath,inputJsonPath)
else:
print("<==== please input image file path ====>")
inputFilePath = input()
imgAug_withoutLabels(inputFilePath)
def script():
"""
eg. argList = [('-l', '--labels', 'label files path','store_true')]
"""
argList = [
('-n', '--nolabel', 'image augmentation without labels'),
('-H', '--HELP',
'show specific help informaton about supported method'),
('-X', '--labelImg',
'image augmentation for labelImg, default labelme. "X" for xml'),
# ('--number', 'image augmentation numbers, default 1'),
{
'shortName': '-N',
'fullName': '--number',
'type': int,
'help': 'image augmentation numbers, default 1'
},
('-L', '--nolog', 'remove "annoying" logs'),
{
'shortName':
'-c',
'fullName':
'--classfilepath',
'type':
str,
'help':
'class-information-path(for labelme is a *.yaml file,for labelImg is a *.txt file. without this file, this script has some errors when generate mask files and image augumentation.)'
}
]
p = Parser(argList, __appname__)
parser = p.get_parser()
args = vars(parser.parse_args())
# print(args)
if args['version']:
# print(1)
from convertmask import __version__
print(__version__)
del __version__
return
if not args['method'] and not args['version']:
parser.print_help()
return
if args['method'] and not args['input'] and not args['HELP']:
logger.error('<=== INPUT FOLDER/FILE MUST NOT BE NULL ===>')
return
if args['nolog']:
ccfg.setConfigParam(ccfg.cfp, 'log', 'show', 'False')
if args['method'] not in __support_methods__ and args[
'method'] not in supported_simplified_methods:
# print(args['method'])
kvs = getKV(__support_methods_simplified__)
logger.warning(' only ==>{} are supported'.format('\n==>'.join(kvs)))
lis = difflib.get_close_matches(args['method'], __support_methods__)
# print(lis)
if len(lis) > 0:
logger.info('do you mean: {} ?'.format(' OR '.join(lis)))
del lis
return
if args['method'] == 'mask2json' or args[
'method'] == __support_methods_simplified__['mask2json']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(
' origin-image-path(necessary), mask-image-path(necessary), yaml-file-path(can be blank,better dont)'
)
print(
' origin images are used to generate base64 code. mask images are used to generate polygons. yaml file saves classes information'
)
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 2:
raise MethodInputException('Not enough input parameters')
elif len(params) == 2:
inputOriimgPath = params[0]
inputMaskPath = params[1]
savePath = os.path.dirname(inputMaskPath)
inputYamlPath = ''
elif len(params) == 3:
inputOriimgPath = params[0]
inputMaskPath = params[1]
savePath = os.path.dirname(inputMaskPath)
inputYamlPath = params[2]
else:
raise MethodInputException('Too much input parameters')
getJsons(inputOriimgPath, inputMaskPath, savePath, inputYamlPath)
print('Done!')
if args['method'] == 'mask2xml' or args[
'method'] == __support_methods_simplified__['mask2xml']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(
' origin-image-path(necessary), mask-image-path(necessary)')
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 2:
raise MethodInputException('Not enough input parameters')
elif len(params) == 2:
inputOriimgPath = params[0]
inputMaskPath = params[1]
savePath = os.path.dirname(inputMaskPath)
else:
raise MethodInputException('Too much input parameters')
getXmls(inputOriimgPath, inputMaskPath, savePath)
print('Done!')
if args['method'] == 'json2xml' or args[
'method'] == __support_methods_simplified__['json2xml']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(' json-file-path(necessary)')
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 1:
raise MethodInputException('Not enough input parameters')
elif len(params) == 1:
inputJsonPath = params[0]
else:
raise MethodInputException('Too much input parameters')
j2xConvert(inputJsonPath)
print('Done!')
if args['method'] == 'json2mask' or args[
'method'] == __support_methods_simplified__['json2mask']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(' json-file-path(necessary)')
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 1:
raise MethodInputException('Not enough input parameters')
elif len(params) == 1:
inputJsonPath = params[0]
else:
raise MethodInputException('Too much input parameters')
processor(inputJsonPath)
print('Done!')
if args['method'] == 'augmentation' or args[
'method'] == __support_methods_simplified__['augmentation']:
if not args['number']:
number = 1
else:
number = args['number']
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(
' origin-image-path(necessary), label-file-path(alternative).'
)
print(' --nolabel can be added to just augment images.')
print(
' --labelImg can be added to augment xmls, default jsons.')
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if args['classfilepath']:
classFilePath = args['classfilepath']
else:
classFilePath = ''
if len(params) < 1:
raise MethodInputException('Not enough input parameters')
elif len(params) == 1 and args['nolabel']:
inputFilePath = params[0]
imgAug_withoutLabels(inputFilePath, number)
print('Done!')
elif len(params) == 2:
inputFilePath = params[0]
inputJsonPath = params[1]
if not args['labelImg']:
imgAug_withLabels(inputFilePath,
inputJsonPath,
number,
yamlFilePath=classFilePath)
else:
imgAug_LabelImg(inputFilePath, inputJsonPath, number)
print('Done!')
else:
raise MethodInputException('There must be some errors.')
if args['method'] == 'xml2json' or args[
'method'] == __support_methods_simplified__['xml2json']:
logger.info('<=== This is a test function. ===>')
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(' xml-file-path(necessary), origin-image-path(necessary)')
print(' currently, only single file conversion supported.')
print(
' --labelImg can be added which means xmls are generated by labelImg.'
)
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 2:
raise MethodInputException('Not enough input parameters')
elif len(params) == 2:
inputFilePath = params[0]
inputJsonPath = params[1]
if not args['labelImg']:
x2jConvert(inputFilePath, inputJsonPath)
else:
x2jConvert_pascal(inputFilePath, inputJsonPath)
print('Done!')
else:
raise MethodInputException('Too much input parameters')
if args['method'] == 'yolo2xml' or args[
'method'] == __support_methods_simplified__['yolo2xml']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(
' txt-file-path(necessary), origin-image-path(necessary), class-file-path(necessary'
)
print(' currently, only single file conversion supported.')
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 3:
raise MethodInputException('Not enough input parameters')
elif len(params) == 3:
txtpath = params[0]
imgpath = params[1]
labelPath = params[2]
else:
raise MethodInputException('Too much input parameters')
y2xConvert(txtPath=txtpath, imgPath=imgpath, labelPath=labelPath)
print('Done!')
if args['method'] == 'xml2yolo' or args[
'method'] == __support_methods_simplified__['xml2yolo']:
if args['HELP']:
print('\n')
print("<==== {} detailed information ====>".format(args['method']))
print('\n')
print('This method parameter list should follow this order.')
print(' xml-file-path(necessary), class-file-path(alternative)')
print(
' class-file is a txt file saves class information. without this file, a txt file will be generated automaticly, which may be in different orders'
)
print('\n')
print('<==== The End ====>')
print('\n')
else:
params = args['input']
if len(params) < 1:
raise MethodInputException('Not enough input parameters')
elif len(params) == 1:
xmlpath = params[0]
x2yConvert(xmlpath)
print('Done!')
elif len(params) == 2:
xmlpath = params[0]
labelpath = params[1]
x2yConvert(xmlpath, labelpath)
print('Done!')
else:
raise MethodInputException('Too much input parameters')
version: beta
Author: xiaoshuyui
Date: 2020-10-12 10:28:37
LastEditors: xiaoshuyui
LastEditTime: 2020-10-21 16:18:09
'''
import sys
sys.path.append('..')
import os
# from convertmask.utils.imgAug import imgFlip, imgNoise, imgRotation, imgTranslation, aug_labelme, aug_labelimg
from convertmask.utils.imgAug_script import imgAug_LabelImg, imgAug_withLabels
BASE_DIR = os.path.abspath(os.path.dirname(os.getcwd())) + os.sep + 'static'
# print(BASE_DIR)
imgPath = BASE_DIR + os.sep + 'multi_objs.jpg'
labelPath = BASE_DIR + os.sep + 'multi_objs.json'
imgPath2 = BASE_DIR + os.sep + 'label_255.png'
labelPath2 = BASE_DIR + os.sep + 'label_255.xml'
yamlFilePath = BASE_DIR + os.sep + 'multi_objs.yaml'
txtFilePath = BASE_DIR + os.sep + 'multi_objs.txt'
if __name__ == "__main__":
imgAug_withLabels(imgPath, labelPath, number=1, yamlFilePath=yamlFilePath)
# imgAug_LabelImg(imgPath2, labelPath2, number=2, yamlFilePath=txtFilePath)
'''
lanhuage: python
Descripttion: multiple image augmentation script.
version: beta
Author: xiaoshuyui
Date: 2020-10-12 10:28:37
LastEditors: xiaoshuyui
LastEditTime: 2020-10-20 09:45:21
'''
import sys
sys.path.append('..')
import os
# from convertmask.utils.imgAug import imgFlip, imgNoise, imgRotation, imgTranslation, aug_labelme, aug_labelimg
from convertmask.utils.imgAug_script import imgAug_withLabels, imgAug_LabelImg
BASE_DIR = os.path.abspath(os.path.dirname(os.getcwd())) + os.sep + 'static'
# print(BASE_DIR)
imgPath = BASE_DIR + os.sep + 'multi_objs.jpg'
labelPath = BASE_DIR + os.sep + 'multi_objs.json'
imgPath2 = BASE_DIR + os.sep + 'label_255.png'
labelPath2 = BASE_DIR + os.sep + 'label_255.xml'
if __name__ == "__main__":
imgAug_withLabels(imgPath, labelPath, number=2)
imgAug_LabelImg(imgPath2, labelPath2, number=50)