Пример #1
0
def main(sysargs):
    args = EasyArgs(sysargs)
    cfg = EasyConfig(args.cfg, group="compare_trainer")
    window_name = "EagleEye Comparator"
    
    if "help" in args:
        usage()
        return 0
    
    # grab marks from args
    if len(args) > 5:
        mark_in = args[4]
        mark_out = args[5]
        
        # test integer-ness
        try: int(mark_in) and int(mark_out)
        except: 
            usage()
            return 1
    
    # or grab them from a marker tool
    elif len(args) > 3:
        ret, mark_in, mark_out = marker_tool(args[1], 50, window_name)
        if not ret:
            print "Not processing - exiting."
            return 1
    else:
        usage()
        return 1
    
    # determine sides
    side = Theta.resolve(args.side or 'buttonside')
    
    # open inouts files
    vid = BuffSplitCap(args[1], buff_max=cfg.buffer_size, side=Theta.Both)
    mapper_xml = Xmlset(args[2], offset=cfg.offset, offmode=Xmlset.offset_mode(cfg.offset_mode))
    trainer_xml = Xmltrainer(args[3], side=side)
    reprojerror_list = {}     # list of reprojection error of all frames
    lastframe = False
    
    # reject mapper_xml if it doesn't contain the trainer_target
    if cfg.trainer_target not in mapper_xml.data(0):
        print "Mapping file must contain training target:", cfg.trainer_target
        print "Target names in mapping file:", mapper_xml.data(0)
        print "Please specifiy correct target name in config"
        return 1
    
    # also reject if it's lens data doesn't match the trainer_xml
    _test_lens = mapper_xml.data(0)[cfg.trainer_target]['lens']
    if (_test_lens == Theta.NonDual and side != Theta.NonDual) \
            or (side != Theta.NonDual and _test_lens == Theta.NonDual):
        print "Mapping file doesn't match training file"
        print "map:", Theta.name(_test_lens), "trainer:", Theta.name(side)
        return 1
    
    # sync the video and xml
    vid.restrict(mark_in, mark_out)
    cropped_total = mark_out - mark_in
    mapper_xml.setRatio(cropped_total)
    
    # status
    print "ratio at:", mapper_xml.ratio()
    print "offset by:", cfg.offset, "in", cfg.offset_mode, "mode"
    print ""
    
    # open export (if specified)
    if args.video_export:
        in_fps  = vid.get(cv2.CAP_PROP_FPS)
        in_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
                   int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))

        out_vid = cv2.VideoWriter(args.video_export,
                                  cv2.VideoWriter_fourcc(*cfg.fourcc),
                                  in_fps, vid.shape[:2])
        print "exporting to:", args.video_export
    else:
        cv2.namedWindow(window_name)
        print "interactive mode"
    
    
    # main loop
    while vid.isOpened():
        frame = vid.frame()
        reprojerror_list = compareReproj(frame, vid.at(), mapper_xml, trainer_xml, reprojerror_list, cfg)
        
        # export or navigate
        if args.video_export:
            sys.stdout.write("Reading Video Frame: {} / {}\r".format(vid.at(), mark_out))
            sys.stdout.flush()
            
            out_vid.write(frame)
            
            if vid.next():
                reprojerror_list = compareReproj(frame, vid.at(), mapper_xml, trainer_xml, reprojerror_list, cfg)
                mapper_xml.next()
            else:
                calReprojList(reprojerror_list)
                print "\nend of video"
                break
        else:
            cv2.imshow(window_name, frame)
            key = cv2.waitKey(0)
            showAll = False     # toggle to show all training points in one frame
                
            # controls
            if key == Key.esc:
                print "\nexiting."
                break
            elif key == Key.right:
                if vid.next():
                    mapper_xml.next()
                else:
                    if lastframe is False:
                        print "\nEnd of video"
                        calReprojList(reprojerror_list)
                    lastframe = True
                    
            elif key == Key.left:
                if vid.back():
                    mapper_xml.back()
            
            elif key == Key.enter:
                while True:
                    reprojerror_list = compareReproj(frame, vid.at(), mapper_xml, trainer_xml, reprojerror_list, cfg)
                    cv2.imshow(window_name, frame)
                    
                    if vid.next():
                        mapper_xml.next()
                        frame = vid.frame()
                    
                    # break when over
                    else: break
                
                calReprojList(reprojerror_list)
                print "\nFast forwarded to end of video"
                
                if args.compare_export:
                    print "Output to file"
                    writeFile(args.compare_export, reprojerror_list)
                break
            
            elif key == Key.space:
                ''' TODO: display all training points, may be abandoned
                problems:
                frame is not refreshed after imshow
                if showAll is False:
                    displayTrainPts(frame, mark_in, mark_out, trainer_xml, cfg)
                    #
                    #except:
                    #    print "Error in displaying training points. Exiting..."
                    #    return 1
                    #
                    
                    showAll = True
                else:
                    frame, reprojerror_list = compareReproj(frame, vid.at(), mapper_xml, trainer_xml, reprojerror_list, cfg)
                    showAll = False
                '''
                pass
    
    # clean up
    vid.release()
    if args.video_export:
        out_vid.release()
    else:
        cv2.destroyAllWindows()
        
    return 0
Пример #2
0
def main(sysargs):
    args = EasyArgs(sysargs)
    cfg = EasyConfig(args.cfg, group="compare")
    window_name = "EagleEye Comparator"
    
    if "help" in args:
        usage()
        return 0
    
    # grab marks from args
    if len(args) > 5:
        mark_in = args[4]
        mark_out = args[5]
        
        # test integer-ness
        try: int(mark_in) and int(mark_out)
        except: 
            usage()
            return 1
    
    # or grab them from a marker tool
    elif len(args) > 3:
        ret, mark_in, mark_out = marker_tool(args[1], 50, window_name)
        if not ret:
            print "Not processing - exiting."
            return 1
    else:
        usage()
        return 1
    
    # open video files
    vid = BuffSplitCap(args[1], buff_max=cfg.buffer_size)
    xml1 = Xmlset(args[2])
    xml2 = Xmlset(args[3])
    
    # trim the video
    vid.restrict(mark_in, mark_out)
    
    # trim the CSV
    cropped_total = mark_out - mark_in
    xml1.setRatio(cropped_total)
    xml2.setRatio(cropped_total) # should be 1.0 (total frames should be the same as video)
    print 'xml1 ratio:', xml1.ratio()
    print 'xml2 ratio:', xml2.ratio()
    
    # open export (if specified)
    if 'export' in args:
        in_fps  = vid.get(cv2.CAP_PROP_FPS)
        in_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
                   int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))

        out_vid = cv2.VideoWriter(args.export,
                                  cv2.VideoWriter_fourcc(*cfg.fourcc),
                                  in_fps, in_size)
    else:
        cv2.namedWindow(window_name)
    
    while vid.isOpened():
        sys.stdout.write(vid.status() + "\r")
        sys.stdout.flush()
        
        frame = vid.frame()
        
        # draw objects from each dataset
        for name in xml1.data():
            draw(frame, xml1.data()[name], cfg.xml1_colour)
        for name in xml2.data():
            draw(frame, xml2.data()[name], cfg.xml2_colour)
        
        # print status to screen
        displayText(frame, vid.status(), top=True)
        displayText(frame, "{}: {}".format(os.path.basename(args[2]), xml1.status()), colour=cfg.xml1_colour)
        displayText(frame, "{}: {}".format(os.path.basename(args[3]), xml2.status()), colour=cfg.xml2_colour)
        
        # export or navigate
        if 'export' in args:
            sys.stdout.write("{}/{}\r".format(vid.at(), cropped_total))
            sys.stdout.flush()
            
            out_vid.write(frame)
            
            if vid.next():
                xml1.next()
                xml2.next()
            else:
                print "\nend of video"
                break
        else:
            cv2.imshow(window_name, frame)
            key = cv2.waitKey(0)
            
            # controls
            if key == Key.esc:
                print "\nexiting."
                break
            elif key == Key.right:
                if vid.next():
                    xml1.next()
                    xml2.next()
            elif key == Key.left:
                if vid.back():
                    xml1.back()
                    xml2.back()
    
    # clean up
    vid.release()
    if 'export' in args:
        out_vid.release()
    else:
        cv2.destroyAllWindows()
    
    return 0
Пример #3
0
def main(sysargs):
    args = EasyArgs(sysargs)
    cfg = EasyConfig(args.cfg, group="compare")
    window_name = "EagleEye Comparator"

    if "help" in args:
        usage()
        return 0

    # grab marks from args
    if len(args) > 5:
        mark_in = args[4]
        mark_out = args[5]

        # test integer-ness
        try:
            int(mark_in) and int(mark_out)
        except:
            usage()
            return 1

    # or grab them from a marker tool
    elif len(args) > 3:
        ret, mark_in, mark_out = marker_tool(args[1], 50, window_name)
        if not ret:
            print "Not processing - exiting."
            return 1
    else:
        usage()
        return 1

    # open video files
    vid = BuffSplitCap(args[1], buff_max=cfg.buffer_size)
    xml1 = Xmlset(args[2])
    xml2 = Xmlset(args[3])

    # trim the video
    vid.restrict(mark_in, mark_out)

    # trim the CSV
    cropped_total = mark_out - mark_in
    xml1.setRatio(cropped_total)
    xml2.setRatio(cropped_total
                  )  # should be 1.0 (total frames should be the same as video)
    print 'xml1 ratio:', xml1.ratio()
    print 'xml2 ratio:', xml2.ratio()

    # open export (if specified)
    if 'export' in args:
        in_fps = vid.get(cv2.CAP_PROP_FPS)
        in_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
                   int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))

        out_vid = cv2.VideoWriter(args.export,
                                  cv2.VideoWriter_fourcc(*cfg.fourcc), in_fps,
                                  in_size)
    else:
        cv2.namedWindow(window_name)

    while vid.isOpened():
        sys.stdout.write(vid.status() + "\r")
        sys.stdout.flush()

        frame = vid.frame()

        # draw objects from each dataset
        for name in xml1.data():
            draw(frame, xml1.data()[name], cfg.xml1_colour)
        for name in xml2.data():
            draw(frame, xml2.data()[name], cfg.xml2_colour)

        # print status to screen
        displayText(frame, vid.status(), top=True)
        displayText(frame,
                    "{}: {}".format(os.path.basename(args[2]), xml1.status()),
                    colour=cfg.xml1_colour)
        displayText(frame,
                    "{}: {}".format(os.path.basename(args[3]), xml2.status()),
                    colour=cfg.xml2_colour)

        # export or navigate
        if 'export' in args:
            sys.stdout.write("{}/{}\r".format(vid.at(), cropped_total))
            sys.stdout.flush()

            out_vid.write(frame)

            if vid.next():
                xml1.next()
                xml2.next()
            else:
                print "\nend of video"
                break
        else:
            cv2.imshow(window_name, frame)
            key = cv2.waitKey(0)

            # controls
            if key == Key.esc:
                print "\nexiting."
                break
            elif key == Key.right:
                if vid.next():
                    xml1.next()
                    xml2.next()
            elif key == Key.left:
                if vid.back():
                    xml1.back()
                    xml2.back()

    # clean up
    vid.release()
    if 'export' in args:
        out_vid.release()
    else:
        cv2.destroyAllWindows()

    return 0
Пример #4
0
def main(sysargs):
    args = EasyArgs(sysargs)
    cfg = EasyConfig(args.cfg, group="compare_trainer")
    window_name = "EagleEye Comparator"

    if "help" in args:
        usage()
        return 0

    # grab marks from args
    if len(args) > 5:
        mark_in = args[4]
        mark_out = args[5]

        # test integer-ness
        try:
            int(mark_in) and int(mark_out)
        except:
            usage()
            return 1

    # or grab them from a marker tool
    elif len(args) > 3:
        ret, mark_in, mark_out = marker_tool(args[1], 50, window_name)
        if not ret:
            print "Not processing - exiting."
            return 1
    else:
        usage()
        return 1

    # determine sides
    side = Theta.resolve(args.side or 'buttonside')

    # open inouts files
    vid = BuffSplitCap(args[1], buff_max=cfg.buffer_size, side=Theta.Both)
    mapper_xml = Xmlset(args[2],
                        offset=cfg.offset,
                        offmode=Xmlset.offset_mode(cfg.offset_mode))
    trainer_xml = Xmltrainer(args[3], side=side)
    reprojerror_list = {}  # list of reprojection error of all frames
    lastframe = False

    # reject mapper_xml if it doesn't contain the trainer_target
    if cfg.trainer_target not in mapper_xml.data(0):
        print "Mapping file must contain training target:", cfg.trainer_target
        print "Target names in mapping file:", mapper_xml.data(0)
        print "Please specifiy correct target name in config"
        return 1

    # also reject if it's lens data doesn't match the trainer_xml
    _test_lens = mapper_xml.data(0)[cfg.trainer_target]['lens']
    if (_test_lens == Theta.NonDual and side != Theta.NonDual) \
            or (side != Theta.NonDual and _test_lens == Theta.NonDual):
        print "Mapping file doesn't match training file"
        print "map:", Theta.name(_test_lens), "trainer:", Theta.name(side)
        return 1

    # sync the video and xml
    vid.restrict(mark_in, mark_out)
    cropped_total = mark_out - mark_in
    mapper_xml.setRatio(cropped_total)

    # status
    print "ratio at:", mapper_xml.ratio()
    print "offset by:", cfg.offset, "in", cfg.offset_mode, "mode"
    print ""

    # open export (if specified)
    if args.video_export:
        in_fps = vid.get(cv2.CAP_PROP_FPS)
        in_size = (int(vid.get(cv2.CAP_PROP_FRAME_WIDTH)),
                   int(vid.get(cv2.CAP_PROP_FRAME_HEIGHT)))

        out_vid = cv2.VideoWriter(args.video_export,
                                  cv2.VideoWriter_fourcc(*cfg.fourcc), in_fps,
                                  vid.shape[:2])
        print "exporting to:", args.video_export
    else:
        cv2.namedWindow(window_name)
        print "interactive mode"

    # main loop
    while vid.isOpened():
        frame = vid.frame()
        reprojerror_list = compareReproj(frame, vid.at(), mapper_xml,
                                         trainer_xml, reprojerror_list, cfg)

        # export or navigate
        if args.video_export:
            sys.stdout.write("Reading Video Frame: {} / {}\r".format(
                vid.at(), mark_out))
            sys.stdout.flush()

            out_vid.write(frame)

            if vid.next():
                reprojerror_list = compareReproj(frame, vid.at(), mapper_xml,
                                                 trainer_xml, reprojerror_list,
                                                 cfg)
                mapper_xml.next()
            else:
                calReprojList(reprojerror_list)
                print "\nend of video"
                break
        else:
            cv2.imshow(window_name, frame)
            key = cv2.waitKey(0)
            showAll = False  # toggle to show all training points in one frame

            # controls
            if key == Key.esc:
                print "\nexiting."
                break
            elif key == Key.right:
                if vid.next():
                    mapper_xml.next()
                else:
                    if lastframe is False:
                        print "\nEnd of video"
                        calReprojList(reprojerror_list)
                    lastframe = True

            elif key == Key.left:
                if vid.back():
                    mapper_xml.back()

            elif key == Key.enter:
                while True:
                    reprojerror_list = compareReproj(frame, vid.at(),
                                                     mapper_xml, trainer_xml,
                                                     reprojerror_list, cfg)
                    cv2.imshow(window_name, frame)

                    if vid.next():
                        mapper_xml.next()
                        frame = vid.frame()

                    # break when over
                    else:
                        break

                calReprojList(reprojerror_list)
                print "\nFast forwarded to end of video"

                if args.compare_export:
                    print "Output to file"
                    writeFile(args.compare_export, reprojerror_list)
                break

            elif key == Key.space:
                ''' TODO: display all training points, may be abandoned
                problems:
                frame is not refreshed after imshow
                if showAll is False:
                    displayTrainPts(frame, mark_in, mark_out, trainer_xml, cfg)
                    #
                    #except:
                    #    print "Error in displaying training points. Exiting..."
                    #    return 1
                    #
                    
                    showAll = True
                else:
                    frame, reprojerror_list = compareReproj(frame, vid.at(), mapper_xml, trainer_xml, reprojerror_list, cfg)
                    showAll = False
                '''
                pass

    # clean up
    vid.release()
    if args.video_export:
        out_vid.release()
    else:
        cv2.destroyAllWindows()

    return 0