GTimothee

import argparse

def manage_args():
    parser = argparse.ArgumentParser()
    #positional args
    parser.add_argument("input_img", help="input image which mask has been computed", type=str)
    parser.add_argument("input_mask", help="semantic segmentation mask computed from the input image", type=str)
    parser.add_argument("output_dir", help="output directory where the result will be saved", type=str)
    
    '''

GTimothee

if __name__ == "__main__":
    with open('somefile.txt', 'a') as the_file:
        the_file.write('Hello\n')

GTimothee

base = os.path.basename(imagePath)
fileName = os.path.splitext(base)[0]

os.path.isdir(dataPath)
shutil.copyfile(imagePath, newImgPath)
os.makedirs(datasetPath)

if os.path.isdir(datasetPath):
  shutil.rmtree(datasetPath, ignore_errors=True)

GTimothee

try:
        os.chdir(datasetPath)
        zipf = zipfile.ZipFile('./../pascal_voc_dataset.zip', 'a', zipfile.ZIP_DEFLATED)
        zipdir('ImageSets/', zipf)
        zipdir('JPEGImages/', zipf)
        zipdir('SegmentationClass/', zipf)
        zipf.close()
        os.chdir('../')
        print('[INFO] Archive successfully built.')
    except:

GTimothee

from os import walk

f = []
for (dirpath, dirnames, filenames) in walk(mypath):
    f.extend(filenames)
    break

GTimothee

img_path = outDirPath + 'images/' + imgName
img = cv2.imread(img_path, cv2.IMREAD_COLOR)

'''FIRST VERSION'''
mask = cv2.cvtColor(mask, cv2.COLOR_RGB2GRAY)
dest = cv2.bitwise_and(img, img, mask = mask)

for coordinates in coordinatesList:
        coords = coordinates[:len(coordinates)-1]
        #ANCIEN

GTimothee

pour faire un clean d'un .sh édité avec un éditeur de texte standard (different de vi)

tr -d '\r' < export.sh > export_clean.sh

GTimothee

from lxml import etree

def getSifts(xmlFilePath):
    doc = etree.parse(xmlFilePath)
    pags = doc.xpath('//PAG')
    pag = pags[0]
    sifts = pag.xpath('SIFTMATCHER')
    return sifts
  
for sift in sifts :

GTimothee

def drawMask(width, height, coordinatesList):
    mask = np.zeros((height,width,3), np.uint8)
    color = (255,255,255)
    for coordinates in coordinatesList:
        cv2.fillConvexPoly(mask, np.array(coordinates, 'int32'), color)
    return mask

GTimothee

https://www.digitalocean.com/community/tutorials/how-to-handle-plain-text-files-in-python-3

#from list of tuples
class Person(object):
    def __init__(self, name, profession):
        self.name = name
        self.profession = profession
 
people = [Person("Nick", "Programmer"), Person("Alice","Engineer")]
professions = dict([ (p.name, p.profession) for p in people ])