【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("未知文件类型")