Merge dev commit '28668532ed2f97a3883be7e68bced252bd1f99bf' into addFeature

wu_xinjun

2022-03-23 563f7741ad28a6a578b56c98c27f92b3677958bd

 src/utils/tools.py

@@ -1,7 +1,9 @@
import os, sys
import numpy as np
import time
from tqdm import tqdm

def file_fliter(folder:str, file_ext:str, sort_format):
def file_fliter(folder:str, file_ext:str, sort_format = None):

    files_List = os.listdir(folder)
    target_files = []
@@ -10,8 +12,10 @@
    for file in files_List:
        if file.split('.')[-1] == file_ext:
            target_files.append(file)
    # sort the list by the sort_format
    target_files.sort(key=sort_format)

    if sort_format is not None:
        # sort the list by the sort_format
        target_files.sort(key=sort_format)

    return target_files

@@ -38,6 +42,29 @@
    return ply_array


def write_ply(data:np.array,folder:str,name:str):
    file = os.path.join(folder,name+".ply")
    with open (file,'w') as f:
        f.write("ply\n")
        f.write("format ascii 1.0\n")
        f.write("comment write_ply generated\n")
        f.write("element vertex %d\n" % data.shape[0])
        f.write("property float x\n")
        f.write("property float y\n")
        f.write("property float z\n")
        f.write("element face 0\n")
        f.write("property list uchar int vertex_indices\n")
        f.write("end_header\n")
        n_p = 0

        for p in tqdm(data):
            x = p[0]
            y = p[1]
            z = p[2]
            f.write("{:.5f} {:.5f} {:.5f}\n".format(x, y, z))
            n_p += 1
        print(time.asctime(),f"Frame {name}.ply have been saved with {n_p} points!")



if __name__ == "__main__":