def initialize(self):
# load settings
# settings = config.settings
self.set_working_dir()
parse_options()
# create app and update settings
init_log()
from app.app import make_app
self.app = make_app(options.COOKIE_SECRET, options.DEBUG)
# send ping to web socket connect users
BeatPing()
self._get_loop()
from lib.heartbeat import heartbeat
self.loop.call_later(0.5, heartbeat.ticker, app=self.app)
self.loop.run_sync(self.app.mq.connect)
self._init_subscribe()
from lib.options import parse_options
from lib.geoutils import sample_unif_sphere, sample_fib_sphere, normalized_slice
def write_exr(path, data):
pyexr.write(path, data,
channel_names={'normal': ['X', 'Y', 'Z'],
'x': ['X', 'Y', 'Z'],
'view': ['X', 'Y', 'Z']},
precision=pyexr.HALF)
if __name__ == '__main__':
# Parse
parser = parse_options(return_parser=True)
app_group = parser.add_argument_group('app')
app_group.add_argument('--img-dir', type=str, default='_results/render_app/imgs',
help='Directory to output the rendered images')
app_group.add_argument('--render-2d', action='store_true',
help='Render in 2D instead of 3D')
app_group.add_argument('--exr', action='store_true',
help='Write to EXR')
app_group.add_argument('--r360', action='store_true',
help='Render a sequence of spinning images.')
app_group.add_argument('--rsphere', action='store_true',
help='Render around a sphere.')
app_group.add_argument('--nb-poses', type=int, default=64,
help='Number of poses to render for sphere rendering.')
app_group.add_argument('--cam-radius', type=float, default=4.0,
help='Camera radius to use for sphere rendering.')
import argparse
import glob
import logging as log
from lib.trainer import Trainer
from lib.options import parse_options
# Set logger display format
log.basicConfig(format='[%(asctime)s] [INFO] %(message)s',
datefmt='%d/%m %H:%M:%S',
level=log.INFO)
if __name__ == "__main__":
"""Main program."""
args, args_str = parse_options()
log.info(f'Parameters: \n{args_str}')
if args.analytic:
for filename in glob.glob(os.path.join(args.glsl_path,
'*/*.glsl'))[::-1]:
name = os.path.basename(filename).split('.')[0]
if args.exclude and name in args.exclude:
continue
if args.include and name not in args.include:
continue
log.info(f'Training {name}')
model = Trainer(args, args_str, name)
model.train()
else:
#coding=utf-8
import pdb , platform
import sys , os , pdb
BASE_DIR = os.path.abspath(__file__)
_root = os.path.dirname(BASE_DIR)
sys.path.append(_root)
import tornado.httpserver
import tornado.ioloop
import tornado.options
import app
import lib.options
from lib.options import parse_options
from tornado.options import define, options
if __name__ == "__main__":
parse_options()
http_server = tornado.httpserver.HTTPServer(app.Application())
http_server.listen(options.SERVER_PORT)
print "\nserver start ! \n"
tornado.ioloop.IOLoop.instance().start()