reorganize.py 900 B

12345678910111213141516171819202122232425262728
  1. import os
  2. import shutil
  3. # 定义目录结构
  4. dirs = {
  5. 'src/core': ['inference.py', 'feature_extractor.py', 'image_validation.py', 'ali_image_validation.py'],
  6. 'src/utils': ['bbox_visualizer.py', 'detection_visualizer.py', 'split_dataset.py', 'update_labels.py'],
  7. 'src/analysis': ['hotspot_analyzer.py', 'report_generator.py', 'analysis_report.py'],
  8. 'data/models': ['UAV-250321.onnx'],
  9. 'data/false_positive': [],
  10. 'tests': [],
  11. 'outputs/archive': []
  12. }
  13. # 创建目录结构
  14. for dir_path in dirs.keys():
  15. os.makedirs(dir_path, exist_ok=True)
  16. # 移动文件
  17. for target_dir, files in dirs.items():
  18. for file in files:
  19. if os.path.exists(file):
  20. shutil.move(file, os.path.join(target_dir, file))
  21. print(f'Moved {file} to {target_dir}/')
  22. else:
  23. print(f'Warning: {file} not found')
  24. print('Project reorganization completed!')