File size: 661 Bytes
a3a407d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | 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()
|