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")))