def get_args(path):
'''
Parse command line args & override defaults.
Arguments:
- path (str) Absolute filepath
Returns:
- (tuple) Name, email, license, project, ext, year
'''
defaults = get_defaults(path)
licenses = ', '.join(os.listdir(cwd + licenses_loc))
p = parser(description='tool for adding open source licenses to your projects. available licenses: %s' % licenses)
_name = False if defaults.get('name') else True
_email = False if defaults.get('email') else True
_license = False if defaults.get('license') else True
p.add_argument('-n', dest='name', required=_name, help='name')
p.add_argument('-e', dest='email', required=_email, help='email')
p.add_argument('-l', dest='license', required=_license, help='license')
p.add_argument('-p', dest='project', required=False, help='project')
p.add_argument('--txt', action='store_true', required=False, help='add .txt to filename')
args = p.parse_args()
name = args.name if args.name else defaults.get('name')
email = args.email if args.email else defaults.get('email')
license = get_license(args.license, licenses_loc=licenses_loc) if args.license else defaults.get('license')
project = args.project if args.project else os.getcwd().split('/')[-1]
ext = '.txt' if args.txt else ''
year = str(date.today().year)
return (name, email, license, project, ext, year)
def init():
global M_dm, M_disk, M_bulge, M_gas
global N_dm, N_disk, N_bulge, N_gas
global a_dm, a_bulge, Rd, z0
global N_CORES, output
global N_temp, N_range
flags = parser(description="Description.")
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
flags.add_argument('-cores', help='The number of cores to use.',
default=1)
args = flags.parse_args()
output = args.o
N_CORES = int(args.cores)
if not (path.isfile("header.txt") and path.isfile("galaxy_param.txt")):
print "header.txt or galaxy_param.txt missing."
exit(0)
vars_ = process_input("galaxy_param.txt")
M_dm, M_disk, M_bulge, M_gas = (float(i[0]) for i in vars_[0:4])
N_dm, N_disk, N_bulge, N_gas = (int(i[0]) for i in vars_[4:8])
a_dm, a_bulge, Rd, z0 = (float(i[0]) for i in vars_[8:12])
N_temp = {'gas': int(1e4), 'dm': int(5e4), 'disk': int(2e4), 'bulge': int(2e4)}
N_range = {'gas': np.linspace(N_temp['gas'], N_gas, 20),
'dm': np.linspace(N_temp['dm'], N_dm, 20),
'disk': np.linspace(N_temp['disk'], N_disk, 20),
'bulge': np.linspace(N_temp['bulge'], N_bulge, 20)}
def add_license():
p = parser(description='quickly add an open-source license to your project')
p.add_argument('-l', dest='license', required=True, help='license to add')
p.add_argument('-e', dest='email', required=True, help='your email address')
p.add_argument('-n', dest='name', required=True, help='your name')
p.add_argument('-p', dest='project', required=True, help='project name')
args = p.parse_args()
author = args.name + ' <' + args.email + '>'
year = str(date.today().year)
project = args.project
try:
license = licenses[args.license]
except KeyError:
p.exit(1, 'fatal: license %s does not exist\n' % args.license)
license = re.sub(r'\[[author]+\]', author, license)
license = re.sub(r'\[[year]+\]', year, license)
license = re.sub(r'\[[project]+\]', project, license)
with open('LICENSE.txt', 'w') as f:
f.write(license)
def init():
global M_halo, M_disk, M_bulge, M_gas
global N_halo, N_disk, N_bulge, N_gas
global a_halo, a_bulge, Rd, z0, z0_gas
global N_total, M_total
global phi_grid, rho_axis, z_axis, N_rho, Nz
global halo_core, bulge_core, N_CORES, force_yes, output, gas, factor, Z
flags = parser(description="Generates an initial conditions file for a\
galaxy simulation with halo, stellar disk,\
gaseous disk and bulge components.")
flags.add_argument('--nogas', help='Generates a galaxy without gas.',
action='store_true')
flags.add_argument('-cores', help='The number of cores to use during the\
potential canculation. Make sure this\
number is a factor of N_rho*N_z.',
default=1)
flags.add_argument('--force-yes', help='Don\'t ask if you want to use the\
existing potential_data.txt file.\
Useful for automating the execution\
of the script.', action='store_true')
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
args = flags.parse_args()
gas = not args.nogas
N_CORES = int(args.cores)
force_yes = args.force_yes
output = args.o
if not (path.isfile("header.txt") and path.isfile("params_galaxy.txt")):
print "header.txt or params_galaxy.txt missing."
exit(0)
vars_ = process_input("params_galaxy.txt")
M_halo, M_disk, M_bulge, M_gas = (float(i[0]) for i in vars_[0:4])
N_halo, N_disk, N_bulge, N_gas = (float(i[0]) for i in vars_[4:8])
a_halo, a_bulge, Rd, z0, z0_gas = (float(i[0]) for i in vars_[8:13])
halo_core, bulge_core = (i[0][:-1] == 'True' for i in vars_[13:15])
factor = float(vars_[15][0])
Z = float(vars_[16][0])
z0_gas *= z0
if not gas:
N_gas = 0
M_gas = 0
M_total = M_disk + M_bulge + M_halo + M_gas
N_total = N_disk + N_bulge + N_halo + N_gas
N_rho = Nz = 256 # Make sure N_CORES is a factor of N_rho*Nz.
phi_grid = np.zeros((N_rho, Nz))
rho_max = 300 * a_halo
# This has to go far so I can estimate the integrals below.
z_max = 3000 * a_halo
rho_axis = np.logspace(-2, log10(rho_max), N_rho)
z_axis = np.logspace(-2, log10(z_max), Nz)
def init():
global dm_core, gas_core
flags = parser(description="Plots stuff.")
flags.add_argument('--dm-core', help='Sets the density profile for the\
dark matter to have a core.', action='store_true')
flags.add_argument('--gas-core', help='The same, but for the bulge.',
action='store_true')
flags.add_argument('i', help='The name of the input file.',
metavar="file.dat")
args = flags.parse_args()
dm_core = args.dm_core
gas_core = args.gas_core
input_ = args.i
return input_
def init():
global gas, dm, output
global M_dm, a_dm, N_dm, M_gas, a_gas, N_gas, Z
global truncation_radius, gamma_gas, gamma_dm
flags = parser(description="Generates an initial conditions file\
for a galaxy cluster halo simulation.")
flags.add_argument('--no-dm',
help='No dark matter particles in the\
initial conditions. The dark matter potential is\
still used when calculating the gas temperatures.',
action='store_true')
flags.add_argument('--no-gas',
help='Gas is completely ignored, and\
only dark matter is included.',
action='store_true')
flags.add_argument('-o',
help='The name of the output file.',
metavar="init.dat",
default="init.dat")
args = flags.parse_args()
output = args.o
if not path.isfile("params_cluster.ini"):
print "params_cluster.ini missing."
exit(0)
if args.no_dm:
if args.no_gas:
print "Neither gas or dark matter were selected!"
exit(0)
else:
gas = True
dm = False
elif args.no_gas:
gas = False
dm = True
else:
gas = True
dm = True
config = ConfigParser()
config.read("params_cluster.ini")
M_dm = config.getfloat('dark_matter', 'M_dm')
a_dm = config.getfloat('dark_matter', 'a_dm')
N_dm = config.getint('dark_matter', 'N_dm')
gamma_dm = config.getfloat('dark_matter', 'gamma_dm')
if (gas):
M_gas = config.getfloat('gas', 'M_gas')
a_gas = config.getfloat('gas', 'a_gas')
N_gas = config.getint('gas', 'N_gas')
Z = config.getfloat('gas', 'Z')
gamma_gas = config.getfloat('gas', 'gamma_gas')
truncation_radius = config.getfloat('global', 'truncation_radius')
def init():
global gas, dm, output
global M_dm, a_dm, N_dm, M_gas, a_gas, N_gas, Z
global truncation_radius, gamma_gas, gamma_dm
flags = parser(description="Generates an initial conditions file\
for a galaxy cluster halo simulation.")
flags.add_argument('--no-dm', help='No dark matter particles in the\
initial conditions. The dark matter potential is\
still used when calculating the gas temperatures.',
action='store_true')
flags.add_argument('--no-gas', help='Gas is completely ignored, and\
only dark matter is included.',
action='store_true')
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
args = flags.parse_args()
output = args.o
if not path.isfile("params_cluster.ini"):
print "params_cluster.ini missing."
exit(0)
if args.no_dm:
if args.no_gas:
print "Neither gas or dark matter were selected!"
exit(0)
else:
gas = True
dm = False
elif args.no_gas:
gas = False
dm = True
else:
gas = True
dm = True
config = ConfigParser()
config.read("params_cluster.ini")
M_dm = config.getfloat('dark_matter', 'M_dm')
a_dm = config.getfloat('dark_matter', 'a_dm')
N_dm = config.getint('dark_matter', 'N_dm')
gamma_dm = config.getfloat('dark_matter', 'gamma_dm')
if(gas):
M_gas = config.getfloat('gas', 'M_gas')
a_gas = config.getfloat('gas', 'a_gas')
N_gas = config.getint('gas', 'N_gas')
Z = config.getfloat('gas', 'Z')
gamma_gas = config.getfloat('gas', 'gamma_gas')
truncation_radius = config.getfloat('global', 'truncation_radius')
def init():
global dm_core, gas_core
flags = parser(description="Plots stuff.")
flags.add_argument('--dm-core',
help='Sets the density profile for the\
dark matter to have a core.',
action='store_true')
flags.add_argument('--gas-core',
help='The same, but for the bulge.',
action='store_true')
flags.add_argument('i',
help='The name of the input file.',
metavar="file.dat")
args = flags.parse_args()
dm_core = args.dm_core
gas_core = args.gas_core
input_ = args.i
return input_
def init():
global gas, dm, gas_core, dm_core, output
global M_dm, a_dm, N_dm, M_gas, a_gas, N_gas
global max_radius
flags = parser(description="Generates an initial conditions file\
for a galaxy cluster halo simulation.")
flags.add_argument('--gas-core', help='Sets the density profile for the\
gas to have a core.', action='store_true')
flags.add_argument('--dm-core', help='The same, but for the dark matter.',
action='store_true')
flags.add_argument('--no-dm', help='No dark matter particles in the\
initial conditions. The dark matter potential is\
still used when calculating the gas temperatures.',
action='store_true')
flags.add_argument('--no-gas', help='No gas, only dark matter.',
action='store_true')
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
args = flags.parse_args()
gas_core = args.gas_core
dm_core = args.dm_core
output = args.o
if not (path.isfile("header.txt") and path.isfile("params_cluster.txt")):
print "header.txt or params_cluster.txt missing."
exit(0)
if args.no_dm:
if args.no_gas:
print "Neither gas or dark matter were selected!"
exit(0)
else:
gas = True
dm = False
elif args.no_gas:
gas = False
dm = True
else:
gas = True
dm = True
vars_ = process_input("params_cluster.txt")
M_dm, a_dm, N_dm = (float(i[0]) for i in vars_[0:3])
if(gas):
M_gas, a_gas, N_gas = (float(i[0]) for i in vars_[3:6])
max_radius = float(vars_[6][0])
def init():
global M_dm, M_disk, M_bulge, M_gas
global N_dm, N_disk, N_bulge, N_gas
global a_dm, a_bulge, Rd, z0
global N_CORES, output
global N_temp, N_range
flags = parser(description="Description.")
flags.add_argument('-o',
help='The name of the output file.',
metavar="init.dat",
default="init.dat")
flags.add_argument('-cores', help='The number of cores to use.', default=1)
args = flags.parse_args()
output = args.o
N_CORES = int(args.cores)
if not (path.isfile("header.txt") and path.isfile("galaxy_param.txt")):
print "header.txt or galaxy_param.txt missing."
exit(0)
vars_ = process_input("galaxy_param.txt")
M_dm, M_disk, M_bulge, M_gas = (float(i[0]) for i in vars_[0:4])
N_dm, N_disk, N_bulge, N_gas = (int(i[0]) for i in vars_[4:8])
a_dm, a_bulge, Rd, z0 = (float(i[0]) for i in vars_[8:12])
N_temp = {
'gas': int(1e4),
'dm': int(5e4),
'disk': int(2e4),
'bulge': int(2e4)
}
N_range = {
'gas': np.linspace(N_temp['gas'], N_gas, 20),
'dm': np.linspace(N_temp['dm'], N_dm, 20),
'disk': np.linspace(N_temp['disk'], N_disk, 20),
'bulge': np.linspace(N_temp['bulge'], N_bulge, 20)
}
def add_license():
p = parser(description='quickly add an open-source license to your project')
p.add_argument('-l', dest='license', required=True, help='license to add')
p.add_argument('-e', dest='email', required=True, help='your email address')
p.add_argument('-n', dest='name', required=True, help='your name')
p.add_argument('-p', dest='project', required=True, help='project name')
p.add_argument('--no', action='store_true', required=False, help='removes file extension')
args = p.parse_args()
author = args.name + ' <' + args.email + '>'
year = str(date.today().year)
project = args.project
ext = '' if args.no else '.txt'
try:
license = open(os.path.join(os.path.abspath("./Licenses/"),args.license)).read()
except KeyError:
p.exit(1, 'fatal: license %s does not exist\n' % args.license)
license = license.format(author=author, year=year, project=project)
with open('LICENSE' + ext, 'w') as f:
f.write(license)
def get_args(path):
'''
Parse command line args & override defaults.
Arguments:
- path (str) Absolute filepath
Returns:
- (tuple) Name, email, license, project, ext, year
'''
defaults = get_defaults(path)
licenses = ', '.join(os.listdir(cwd + licenses_loc))
p = parser(description='tool for adding open source licenses to your projects. available licenses: %s' % licenses)
_name = False if defaults.get('name') else True
_email = False if defaults.get('email') else True
_license = False if defaults.get('license') else True
p.add_argument('-n', dest='name', required=_name, help='name')
p.add_argument('-e', dest='email', required=_email, help='email')
p.add_argument('-l', dest='license', required=_license, help='license')
p.add_argument('-p', dest='project', required=False, help='project')
p.add_argument('-v', '--version', action='version', version='%(prog)s {version}'.format(version=version))
p.add_argument('--txt', action='store_true', required=False, help='add .txt to filename')
args = p.parse_args()
name = args.name if args.name else defaults.get('name')
email = args.email if args.email else defaults.get('email')
license = get_license(args.license) if args.license else defaults.get('license')
project = args.project if args.project else os.getcwd().split('/')[-1]
ext = '.txt' if args.txt else ''
year = str(date.today().year)
return (name, email, license, project, ext, year)
sys.path.append('/home/beams/CCHUANG/codezoo/python/img_util/')
import img_util as imu
import tomopy
import dxchange.reader as dxreader
import dxchange.writer as dxwriter
import numpy as np
import time
import os
import cv2
from argparse import ArgumentParser as parser
if __name__ == '__main__':
# create parser
par = parser(description="Tomography Image Reconstruction Script")
# define input
par.add_argument("exp_log", help="Exp_log file")
par.add_argument("scan_id",
help="ID of the scan (defined in Exp_log file)")
par.add_argument("rec_type", help="'full' or 'sub' reconstruction")
# define optional input
par.add_argument("--rot-axis", type=float, help="specify rotation axis")
par.add_argument("--rot-step", type=float, help="rotation axis step(s)")
par.add_argument("--mblur", type=int, help="ksize of median filter")
args = par.parse_args() # returns data from the options specified (echo)
# initialize parameters
opt = imu.tomo_par_init(args.exp_log, args.scan_id)
try:
log("Update stack called","INFO")
response = client.update_stack(
StackName = stack_name,
TemplateBody = template_body,
Parameters=[params]
#TimeoutInMinutes=30,
#Capabilities=[capabilities],
#OnFailure='ROLLBACK',
#EnableTerminationProtection=True
)
except Exception as e:
log(e,"Error")
arg = parser(description='Get additional arguments')
arg.add_argument("-op","--operation",help="operation", dest="operation", required=False)
args = arg.parse_args()
op = args.operation
config = config.ConfigParser()
config.optionxform=str
config.read("config.ini")
options = config.items("production-stack-params")
params = {}
for k,v in options:
from sys import platform
from os import getenv, listdir, remove
from shutil import copyfile
from argparse import ArgumentParser as parser
from re import sub
if platform.startswith('linux') or platform == 'darwin':
print("Linux and macOS aren't supported yet!")
exit()
print('Searching for game install...')
ar_parser = parser()
ar_parser.add_argument('-u',
'--uninstall',
help='uninstall hack',
action='store_true',
default=False)
args = ar_parser.parse_args()
supported_versions = ['3.1.35', '3.1.36', '3.1.37']
appdata = getenv('APPDATA')
if not appdata:
print(
"I couldn't find where apps are installed on the system. Is this a modded Windows install?"
)
exit()
folders = listdir(appdata)
#!/usr/bin/env python3
"""
TODO
"""
import argparse
from twitch.modules import *
from twitch import api
p = argparse.parser(description='Description')
p.parse_args()
if __name__ == '__main__':
print('blurb')
agent_dict["n_games"],
agent_dict["n_episodes"],
alpha=agent_dict["alpha"],
gamma=agent_dict["gamma"],
epsilon=agent_dict["epsilon"],
lambd=agent_dict["lambd"],
n_step=agent_dict["n_step"]
)
test_result = dict_result["tests_result"]
return test_result
if __name__ == '__main__':
parser = parser(prog='Saving_data_multiprocess', \
description='Reinforcement learning training agent')
parser.add_argument('-a', '--automatic', action='store_true', \
help='Generate automatically test for agent in enviroment')
args = parser.parse_args()
agents_list = []
tests_result = []
create_custom_enviroment()
enviroment_name = input("Insert the enviroment name: ")
enviroment = gym.make(enviroment_name) #Creazione ambiente
tests_moment = input("Select the test type (final, on_run, ten_perc): ")
#Generazione automatica degli agenti
if args.automatic:
def init():
global M_halo, M_disk, M_bulge, M_gas
global N_halo, N_disk, N_bulge, N_gas
global a_halo, a_bulge, gamma_halo, gamma_bulge, Rd, z0, z0_gas
global halo_cut_r, halo_cut_M, bulge_cut_r, bulge_cut_M
global disk_cut_r, disk_cut
global N_total, M_total
global phi_grid, rho_axis, z_axis, N_rho, Nz
global N_CORES, force_yes, force_no, output, input_, gas, bulge, factor, Z
global file_format
flags = parser(description="Generates an initial conditions file for a\
galaxy simulation with halo, stellar disk,\
gaseous disk and bulge components.")
flags.add_argument('-cores', help='The number of cores to use during the\
potential canculation. Make sure this\
number is a factor of N_rho*N_z. Default\
is 1.',
default=1)
flags.add_argument('--force-yes', help='Don\'t ask if you want to use the\
existing potential_data.txt file.\
Useful for automating the execution\
of the script.', action='store_true')
flags.add_argument('--force-no', help='Same as above, but with the opposite\
effect.', action='store_true')
flags.add_argument('--hdf5', help='Output initial conditions in HDF5\
format.',
action = 'store_true')
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
flags.add_argument('-i', help='The name of the .ini file.',
metavar="params_galaxy.ini", default="params_galaxy.ini")
args = flags.parse_args()
N_CORES = int(args.cores)
force_yes = args.force_yes
force_no = args.force_no
output = args.o
input_ = args.i
if args.hdf5:
file_format = 'hdf5'
else:
file_format = 'gadget2'
if not path.isfile(input_):
print "Input file not found:", input_
exit(0)
config = ConfigParser()
config.read(input_)
# Halo
M_halo = config.getfloat('halo', 'M_halo')
a_halo = config.getfloat('halo', 'a_halo')
N_halo = config.getint('halo', 'N_halo')
gamma_halo = config.getfloat('halo', 'gamma_halo')
halo_cut_r = config.getfloat('halo', 'halo_cut_r')
# Disk
M_disk = config.getfloat('disk', 'M_disk')
N_disk = config.getint('disk', 'N_disk')
Rd = config.getfloat('disk', 'Rd')
z0 = config.getfloat('disk', 'z0')
factor = config.getfloat('disk', 'factor')
disk_cut_r = config.getfloat('disk', 'disk_cut_r')
# Bulge
bulge = config.getboolean('bulge', 'include')
M_bulge = config.getfloat('bulge', 'M_bulge')
a_bulge = config.getfloat('bulge', 'a_bulge')
N_bulge = config.getint('bulge', 'N_bulge')
gamma_bulge = config.getfloat('bulge', 'gamma_bulge')
bulge_cut_r = config.getfloat('bulge', 'bulge_cut_r')
# Gas
gas = config.getboolean('gas', 'include')
M_gas = config.getfloat('gas', 'M_gas')
N_gas = config.getint('gas', 'N_gas')
z0_gas = config.getfloat('gas', 'z0_gas')
Z = config.getfloat('gas', 'Z')
z0_gas *= z0
if not gas:
N_gas = 0
if not bulge:
N_bulge = 0
M_total = M_disk + M_bulge + M_halo + M_gas
N_total = N_disk + N_bulge + N_halo + N_gas
N_rho = config.getint('global', 'N_rho')
Nz = config.getint('global', 'Nz')
phi_grid = np.zeros((N_rho, Nz))
rho_max = config.getfloat('global', 'rho_max')*a_halo
z_max = config.getfloat('global', 'z_max')*a_halo
rho_axis = np.logspace(-2, log10(rho_max), N_rho)
z_axis = np.logspace(-2, log10(z_max), Nz)
def init():
global M_halo, M_disk, M_bulge, M_gas
global N_halo, N_disk, N_bulge, N_gas
global a_halo, a_bulge, Rd, z0, z0_gas
global N_total, M_total
global phi_grid, rho_axis, z_axis, N_rho, Nz
global halo_core, bulge_core, N_CORES, force_yes, output, gas, factor, Z
global file_format
flags = parser(description="Generates an initial conditions file for a\
galaxy simulation with halo, stellar disk,\
gaseous disk and bulge components.")
flags.add_argument('--nogas', help='Generates a galaxy without gas.',
action='store_true')
flags.add_argument('-cores', help='The number of cores to use during the\
potential canculation. Make sure this\
number is a factor of N_rho*N_z. Default\
is 1.',
default=1)
flags.add_argument('--force-yes', help='Don\'t ask if you want to use the\
existing potential_data.txt file.\
Useful for automating the execution\
of the script.', action='store_true')
flags.add_argument('--hdf5', help = 'Generate HDF5 initial conditions. \
Requires h5py.',
action = 'store_true')
flags.add_argument('-o', help='The name of the output file.',
metavar="init.dat", default="init.dat")
args = flags.parse_args()
gas = not args.nogas
N_CORES = int(args.cores)
force_yes = args.force_yes
output = args.o
if args.hdf5:
file_format = 'hdf5'
else:
file_format = 'gadget2'
if not (path.isfile("header.txt") and path.isfile("params_galaxy.ini")):
print "header.txt or params_galaxy.ini missing."
exit(0)
config = ConfigParser()
config.read("params_galaxy.ini")
# Halo
M_halo = config.getfloat('halo', 'M_halo')
a_halo = config.getfloat('halo', 'a_halo')
N_halo = config.getint('halo', 'N_halo')
halo_core = config.getboolean('halo', 'halo_core')
# Disk
M_disk = config.getfloat('disk', 'M_disk')
N_disk = config.getint('disk', 'N_disk')
Rd = config.getfloat('disk', 'Rd')
z0 = config.getfloat('disk', 'z0')
factor = config.getfloat('disk', 'factor')
# Bulge
M_bulge = config.getfloat('bulge', 'M_bulge')
a_bulge = config.getfloat('bulge', 'a_bulge')
N_bulge = config.getint('bulge', 'N_bulge')
bulge_core = config.getboolean('bulge', 'bulge_core')
# Gas
M_gas = config.getfloat('gas', 'M_gas')
N_gas = config.getint('gas', 'N_gas')
z0_gas = config.getfloat('gas', 'z0_gas')
Z = config.getfloat('gas', 'Z')
z0_gas *= z0
if not gas:
N_gas = 0
M_gas = 0
M_total = M_disk + M_bulge + M_halo + M_gas
N_total = N_disk + N_bulge + N_halo + N_gas
N_rho = config.getint('global', 'N_rho')
Nz = config.getint('global', 'Nz')
phi_grid = np.zeros((N_rho, Nz))
rho_max = config.getfloat('global', 'rho_max')*a_halo
z_max = config.getfloat('global', 'z_max')*a_halo
rho_axis = np.logspace(-2, log10(rho_max), N_rho)
z_axis = np.logspace(-2, log10(z_max), Nz)
import numpy as np
import os
import pickle
import matplotlib.pyplot as plt
from argparse import ArgumentParser as parser
from skimage import io
parser = parser(description='Computer Vision 2018 Homework 2 (cnn)')
parser.add_argument('dataset_path',
nargs='?',
default='./hw2-3_data',
help = 'where to find dataset, default: ./hw2-3_data')
args = parser.parse_args()
model_file_name = 'model.model'
data_dir = args.dataset_path + '/train'
x_train = []
y_train = []
count = 0
for i in os.listdir(data_dir):
data_dir_n = data_dir + '/' + i
for j in os.listdir(data_dir_n):
data = io.imread(os.path.join(data_dir_n, j))
x_train.append(np.array(data))
y_train.append(np.array(int(i[6])))
count += 1
x_train = np.array(x_train)
y_train = np.array(y_train)
data_dir = args.dataset_path + '/valid'
def get_arguments(self):
parser = argparse.parser(description="get args")
