test_imgAug.py

'''
@lanhuage: python
@Descripttion: image augmentation with labels.
@version: beta
@Author: xiaoshuyui
@Date: 2020-07-17 15:49:30
LastEditors: xiaoshuyui
LastEditTime: 2020-10-20 09:40:34
'''
import sys
sys.path.append("..")
from convertmask.utils.imgAug import imgFlip, imgNoise, imgRotation, imgTranslation, aug_labelme, aug_labelimg, imgZoom
import os
from skimage import io
from convertmask.utils.getMultiShapes import getMultiShapes

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__":

    #### test1

    # imgFlip(imgPath, labelPath)

    # imgNoise(imgPath,labelPath)

    # imgRotation(imgPath,labelPath)

    # imgTranslation(imgPath,labelPath)

    #### test2

    # n = imgNoise(imgPath,labelPath,flag=False)

    # tmp = n['noise']

    # img , processedImg = tmp.oriImg , tmp.processedImg

    # r = imgRotation(img,processedImg,flag=False,angle=15)

    # tmp = r['rotation']

    # img , processedImg = tmp.oriImg , tmp.processedImg

    # t = imgTranslation(img,processedImg,flag=False)

    # tmp = t['trans']

    # img , processedImg = tmp.oriImg , tmp.processedImg

    # f = imgFlip(img,processedImg,flag=False)

    # tmp = f['h_v']

    # img , processedImg = tmp.oriImg , tmp.processedImg

    # parent_path = os.path.dirname(imgPath)

    # if os.path.exists(parent_path+os.sep+'jsons_'):
    #     pass
    # else:
    #     os.makedirs(parent_path+os.sep+'jsons_')
    # fileName = 'test'
    # io.imsave(parent_path+os.sep+'jsons_'+os.sep+fileName+'_assumble.jpg',img)

    # assumbleJson = getMultiShapes(parent_path+os.sep+'jsons_'+os.sep+fileName+'_assumble.jpg',processedImg,flag=True,labelYamlPath='')

    # saveJsonPath = parent_path+os.sep+'jsons_'+os.sep+fileName+'_assumble.json'

    # with open(saveJsonPath,'w') as f:
    #     f.write(assumbleJson)

    #### test3

    # aug_labelme(imgPath,labelPath)

    #### test4

    # aug_labelimg(imgPath2, labelPath2)

    #### test5
    imgZoom(imgPath, labelPath, 1.2)
    # io.imsave(BASE_DIR + os.sep + 'test_zoom.jpg', i)