Skip to content

Commit

Permalink
Upgrade to new netron lib
Browse files Browse the repository at this point in the history
  • Loading branch information
vkuznet committed Apr 6, 2021
1 parent d78637d commit 14444ca
Show file tree
Hide file tree
Showing 235 changed files with 507,373 additions and 178,192 deletions.
Binary file added src/Go/static/images/favicon.ico
Binary file not shown.
33 changes: 33 additions & 0 deletions src/Go/static/netron/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@

import argparse
import sys
import os

from .server import start
from .server import stop
from .server import wait
from .server import serve
from .__version__ import __version__

def main():
parser = argparse.ArgumentParser(description='Viewer for neural network, deep learning and machine learning models.')
parser.add_argument('file', metavar='MODEL_FILE', help='model file to serve', nargs='?', default=None)
parser.add_argument('-v', '--version', help="print version", action='store_true')
parser.add_argument('-b', '--browse', help='launch web browser', action='store_true')
parser.add_argument('-p', '--port', help='port to serve', type=int)
parser.add_argument('--host', help="host to serve")
parser.add_argument('--log', help='log details to console', action='store_true')
args = parser.parse_args()
if args.file and not os.path.exists(args.file):
print("Model file '" + args.file + "' does not exist.")
sys.exit(2)
if args.version:
print(__version__)
sys.exit(0)
address = (args.host, args.port) if args.host else args.port if args.port else None
start(args.file, address=address, browse=args.browse, log=args.log)
wait()
sys.exit(0)

if __name__ == '__main__':
main()
1 change: 1 addition & 0 deletions src/Go/static/netron/__version__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
__version__ = '0.0.0'
Loading

0 comments on commit 14444ca

Please sign in to comment.