| import shutil | |
| import os | |
| from pathlib import Path | |
| def finalize(): | |
| # Find the latest run | |
| runs_dir = Path("runs/detect/runs") | |
| if not runs_dir.exists(): | |
| print("No runs found.") | |
| return | |
| latest_run = sorted(list(runs_dir.glob("logo_watermark*")), key=os.path.getmtime)[-1] | |
| best_pt = latest_run / "weights" / "best.pt" | |
| if best_pt.exists(): | |
| dst = Path("models/best.pt") | |
| dst.parent.mkdir(parents=True, exist_ok=True) | |
| shutil.copy(best_pt, dst) | |
| print(f"Successfully copied {best_pt} to {dst}") | |
| else: | |
| print(f"Could not find {best_pt}") | |
| if __name__ == "__main__": | |
| finalize() | |