def on_starting(server):
from digits import config
config.load_config()
import argparse
import h5py
import json
import logging
import numpy as np
import os
import sys
# Add path for DIGITS package
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import digits
from digits.config import load_config
load_config()
from digits import utils, log
from digits.inference.errors import InferenceError
from digits.framework_helpers import torch_helpers
logger = logging.getLogger('digits.tools.inference')
"""
Perform inference on a list of images using the specified model
"""
def get_weights(output_dir):
f = h5py.File(os.path.join(output_dir, 'filters.hdf5'), 'w')
vis_file = h5py.File(os.path.join(output_dir, 'weights.h5'), 'r')
def on_starting(server):
from digits import config
config.load_config()