Пример #1
0
def printTrainTestRoidbDictReport(roidbTr, annoCountTr, roidbTe, annoCountTe,
                                  setID, repeat, size):

    numAnnosTr = computeTotalAnnosFromAnnoCount(annoCountTr)
    numAnnosTe = computeTotalAnnosFromAnnoCount(annoCountTe)

    print("\n\n-=-=-=-=-=-=-=-=-\n\n")
    print("Report:\n\n")
    print("Mixture Dataset: {} {} {}\n\n".format(setID, repeat, size))
    print_set_report(roidbTr, numAnnosTr, "train")
    print_set_report(roidbTe, numAnnosTe, "test")
    print("example [train] roidb:")
    for k, v in roidbTr[10].items():
        print("\t==> {},{}".format(k, type(v)))
        print("\t\t{}".format(v))
    printSaveBboxInfo(roidbTr, numAnnosTr, "train")
    printSaveBboxInfo(roidbTe, numAnnosTe, "test")
Пример #2
0
    if args.cfg_file is not None:
        cfg_from_file(args.cfg_file)

    print('Using config:')
    pprint.pprint(cfg)

    if not args.randomize:
        np.random.seed(cfg.RNG_SEED)

    setID = args.setID
    repetition = args.repetition
    size = args.size

    roidb, annoCount = load_mixture_set(setID, repetition, size)
    numAnnos = computeTotalAnnosFromAnnoCount(annoCount)

    print("\n\n-=-=-=-=-=-=-=-=-\n\n")
    print("Report:\n\n")
    print("Mixture Dataset: {} {} {}\n\n".format(setID, repetition, size))

    print("number of images: {}".format(len(roidb)))
    print("number of annotations: {}".format(numAnnos))
    print("size of roidb in memory: {}kB".format(
        len(roidb) * sys.getsizeof(roidb[0]) / 1024.))
    print("example roidb:")
    for k, v in roidb[0].items():
        print("\t==> {},{}".format(k, type(v)))
        print("\t\t{}".format(v))

    print("computing bbox info...")