def arg_params_yolo():
# class YOLO defines the default value, so suppress any default HERE
parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
# Command line options
parser.add_argument('-w',
'--path_weights',
type=str,
help='path to model weight file')
parser.add_argument('-a',
'--path_anchors',
type=str,
help='path to anchor definitions')
parser.add_argument('-c',
'--path_classes',
type=str,
help='path to class definitions')
parser.add_argument('--nb_gpu',
type=int,
help='Number of GPU to use',
default=str(YOLO.get_defaults("nb_gpu")))
parser.add_argument('-o',
'--path_output',
required=False,
type=str,
default='.',
help='path to the output directory')
return parser
def parser():
parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
'''
Command line options
'''
parser.add_argument(
'--model', type=str,
help='path to model weight file, default ' + YOLO.get_defaults("model_path")
)
parser.add_argument(
'--anchors', type=str,
help='path to anchor definitions, default ' + YOLO.get_defaults("anchors_path")
)
parser.add_argument(
'--classes', type=str,
help='path to class definitions, default ' + YOLO.get_defaults("classes_path")
)
parser.add_argument(
'--gpu_num', type=int,
help='Number of GPU to use, default ' + str(YOLO.get_defaults("gpu_num"))
)
parser.add_argument(
'--image', default=True, action="store_true",
help='Image detection mode, will ignore all positional arguments'
)
'''
Command line positional arguments -- for video detection mode
'''
parser.add_argument(
"--input", nargs='?', type=str,required=False,default='./path2your_video',
help = "Video input path"
)
parser.add_argument(
"--output", nargs='?', type=str, default="",
help = "[Optional] Video output path"
)
return parser.parse_args()
def main(args):
ic = Receiver(args)
rospy.init_node("Receiver", anonymous=True)
try:
ic.start()
except KeyboardInterrupt:
print("Shutting down")
print(YOLO.get_defaults("model_path"))
r_image.show()
yolo.close_session()
FLAGS = None
if __name__ == '__main__':
# class YOLO defines the default value, so suppress any default here
parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
'''
Command line options
'''
parser.add_argument('--model',
type=str,
help='path to model weight file, default ' +
YOLO.get_defaults("model_path"))
parser.add_argument('--anchors',
type=str,
help='path to anchor definitions, default ' +
YOLO.get_defaults("anchors_path"))
parser.add_argument('--classes',
type=str,
help='path to class definitions, default ' +
YOLO.get_defaults("classes_path"))
parser.add_argument('--gpu_num',
type=int,
help='Number of GPU to use, default ' +
str(YOLO.get_defaults("gpu_num")))