【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_pat