speed.py 文件源码

def load_dict_from_file(filepath):
        if not os.path.isfile(filepath):
            print("No SpeedSegments database found at file '%s'. Initializing database empty." % (filepath,))
            return dict()
        else:
            lines = open(filepath, "r").readlines()
            #lines = [line.strip().split("\t") for line in lines]
            lines = [line.strip() for line in lines]
            segments = dict()
            nb_found = 0
            nb_labeled = 0
            #segments_found_keys = set()
            #segments_found = []
            for line in lines:
                seg = SpeedSegment.from_string(line)
                segments[seg.get_key()] = seg
                nb_found += 1
                nb_labeled += (seg.label is not None)
            """
            for h, w, vals in lines:
                seg = np.zeros((int(h) * int(w)), dtype=np.bool)
                for i in range(len(vals)):
                    seg[i] = int(vals[i])
                seg = seg.reshape((int(h), int(w)))
                seg_key = SpeedSegment.segment_image_to_key(seg)
                segments[seg_key] = seg
                #segments_found.append(seg)
                #segments_found_keys.add(seg_key)
                nb_found += 1
                nb_labeled += (seg.label is not None)
            #return segments_found_keys, segments_found
            """

            print("Loaded SpeedSegments database from file '%s' with %d segments, %d of which are labeled." % (filepath, nb_found, nb_labeled))
            return segments