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