【python自动创建文件夹】

python自动创建文件夹

python自动创建文件夹，在模型推理时，可自动生成结果文件夹且自动更新，不会覆盖上一次结果，和yolov5中类似；

• 借鉴https://github.com/ultralytics/yolov5/blob/master/utils/general.py中的increment_path函数

• 自动生成新的exp保存文件

  def create_dir(save_dir):     base_name = os.path.basename(save_dir)     cout = int(base_name[3:])     cout+=1     base_name = base_name[:3]+ str(cout)     save_dir = os.path.join(os.path.dirname(save_dir), base_name)    	return save_dir  save_dir = args.out_dir # 自动生成新的保存路径 for _ in range(999):     if os.path.exists(save_dir):         save_dir = create_dir(save_dir)     else:         break if not os.path.exists(save_dir):     os.makedirs(save_dir) config.demo.output_dir = args.output_dir 

• 判断给出给出的路径是文件，还是文件夹

  import os  path = "your_path_here"  if os.path.isfile(path):     print("这是一个文件") elif os.path.isdir(path):     print("这是一个文件夹") else:     print("路径不存在") 

• 判断给出的路径是图像还是一个视频；

  import magic # pip install python-magic  def get_file_type(file_path):     mime = magic.Magic(mime=True)     file_type = mime.from_file(file_path)     return file_type  file_path = "your_file_path_here" file_type = get_file_type(file_path) if "image" in file_type:     print("这是一个图片") elif "video" in file_type:     print("这是一个视频") else:     print("未知文件类型")