Edge-AI-Platform-Tutorials/docs/CATSvsDOGs/set_the_CATSvsDOGS_prj.py at master · andyluo7/Edge-AI-Platform-Tutorials · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# USAGE
# python set_the_CATSvsDOGS_prj.py -i WORK_DIR

# Assumptions:
# 1) you are placed in WORK_DIR and the "train.zip" file is placed there too: for example in my case WORK_DIR == "~/ML/cats-vs-dogs"
# 2) caffe/code subdirectories are already there

# by daniele.bagni@xilinx.com

# ##################################################################################################


# import the necessary packages
import os
import argparse
import glob
import sys


# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--work_dir", required=True, help="path to the working dir")
args = vars(ap.parse_args())

path_root = args["work_dir"] # root path name of dataset


##########################################################################################

if (not os.path.exists(path_root)):
    print("ERROR: the ", path_root, "  directory does not exist!")
    sys.exit(0)

# check the train.zip file is there
if (not os.path.exists(path_root + "/train.zip")):
    print "ERROR: missing train.zip archive !"
    sys.exit(0)

# create directory for input images
if (not os.path.exists(path_root + "/input")):
    os.mkdir(path_root + "/input")

# unzip the train.zip file
os.system("unzip train.zip -d" + path_root + "/input")

# rename "/input/train" as "input/jpg"
os.system("mv " + path_root + "/input/train " + path_root + "/input/jpg")

# create subdirectories "input/jpg/cats" and "input/jpg/dogs"
os.mkdir(path_root + "/input/jpg/cats")
os.mkdir(path_root + "/input/jpg/dogs")

# move all "cat.*.jpg" files into "input/jpg/cats" a
os.system("mv " + path_root + "/input/jpg/cat.*.jpg " + path_root + "/input/jpg/cats")
os.system("mv " + path_root + "/input/jpg/dog.*.jpg " + path_root + "/input/jpg/dogs")


##########################################################################################

## launch python script 1
#command1 = "python " + path_root + "/caffe/code/1_write_cats-vs-dogs_images.py -p " + path_root + "/input/jpg"
#os.system(command1)


## launch python script 2a
#command2 = "python " + path_root + "/caffe/code/2a_create_lmdb.py -i " + path_root + "/input/jpg/ -o " + path_root + "/input/lmdb"
#os.system(command2)


print " END"