def main(args): if not (args.run_train or args.run_test or args.run_reshape or args.download_data): print( 'Set atleast one of the options --train | --test | --reshape | --download-data' ) parser.print_help() return if args.run_test or args.run_train or args.run_reshape: session = get_session(args.gpu_limit) if args.run_train: trainer = HEDTrainer(args.config_file) trainer.setup() trainer.run(session) if args.run_test: tester = HEDTester(args.config_file) tester.setup(session) tester.run(session) if args.run_reshape: reshaper = HEDReshaper(args.config_file) reshaper.setup(session) reshaper.run(session) if args.download_data: io = IO() cfgs = io.read_yaml_file(args.config_file) io.download_data(cfgs['rar_file'], cfgs['download_path'])
def start(self, args): io = IO() self.cfgs = io.read_yaml_file(args.config_file) self.model = Vgg16(self.cfgs, run='testing') meta_model_file = "/Users/csrproject/edge/holy-edge/hed/models/hed-model-5001" saver = tf.train.Saver() session = self.get_session(args.gpu_limit) saver.restore(session, meta_model_file) self.model.setup_testing(session) im = self.fetch_img() edgemap = session.run(self.model.predictions, feed_dict={self.model.images: [im]}) self.save_egdemaps(edgemap)