umyuu commited on
Commit
b76540e
·
1 Parent(s): 447b452

コマンドライン引数の解析を別ファイルに

Browse files
Files changed (2) hide show
  1. app.py +2 -20
  2. src/args_parser.py +23 -0
app.py CHANGED
@@ -2,7 +2,7 @@
2
  """
3
  SaliencyMapDemo
4
  """
5
- from argparse import ArgumentParser, BooleanOptionalAction
6
  #from datetime import datetime
7
  import sys
8
  from typing import Literal
@@ -11,6 +11,7 @@ import gradio as gr
11
  import numpy as np
12
 
13
  from src import PROGRAM_NAME, get_package_version
 
14
  from src.reporter import log
15
  from src.saliency import SaliencyMap, convert_colormap
16
  from src.utils import Stopwatch
@@ -20,25 +21,6 @@ log.info("#アプリ起動中")
20
  watch = Stopwatch.start_new()
21
 
22
 
23
- def parse_args():
24
- """
25
- コマンドライン引数の解析を行います
26
- """
27
- parser = ArgumentParser(prog=PROGRAM_NAME, description=PROGRAM_NAME)
28
- parser.add_argument('--inbrowser',
29
- action=BooleanOptionalAction, default=True, help="Gradio inbrowser")
30
- parser.add_argument('--share',
31
- action=BooleanOptionalAction, default=False, help="Gradio share")
32
- parser.add_argument('--server_port',
33
- type=int, default=7860, help="Gradio server port")
34
- parser.add_argument('--max_file_size',
35
- type=str, default="20MB", help="Gradio max file size")
36
- parser.add_argument('--version',
37
- action='version', version=f'%(prog)s {__version__}')
38
-
39
- return parser.parse_args()
40
-
41
-
42
  def jet_tab_selected(image: np.ndarray):
43
  """
44
  JETタブを選択時
 
2
  """
3
  SaliencyMapDemo
4
  """
5
+
6
  #from datetime import datetime
7
  import sys
8
  from typing import Literal
 
11
  import numpy as np
12
 
13
  from src import PROGRAM_NAME, get_package_version
14
+ from src.args_parser import parse_args
15
  from src.reporter import log
16
  from src.saliency import SaliencyMap, convert_colormap
17
  from src.utils import Stopwatch
 
21
  watch = Stopwatch.start_new()
22
 
23
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  def jet_tab_selected(image: np.ndarray):
25
  """
26
  JETタブを選択時
src/args_parser.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """コマンドライン引数の解析"""
3
+ from argparse import ArgumentParser, BooleanOptionalAction
4
+ from src import PROGRAM_NAME, get_package_version
5
+
6
+
7
+ def parse_args():
8
+ """
9
+ コマンドライン引数の解析を行います
10
+ """
11
+ parser = ArgumentParser(prog=PROGRAM_NAME, description=PROGRAM_NAME)
12
+ parser.add_argument('--inbrowser',
13
+ action=BooleanOptionalAction, default=True, help="Gradio inbrowser")
14
+ parser.add_argument('--share',
15
+ action=BooleanOptionalAction, default=False, help="Gradio share")
16
+ parser.add_argument('--server_port',
17
+ type=int, default=7860, help="Gradio server port")
18
+ parser.add_argument('--max_file_size',
19
+ type=str, default="20MB", help="Gradio max file size")
20
+ parser.add_argument('--version',
21
+ action='version', version=f'%(prog)s {get_package_version()}')
22
+
23
+ return parser.parse_args()