reference_id = "ref_001"
model = torch.hub.load("yolov5", "custom", "Models/id_proof_detection.pt", source="local", verbose=False)
img = Image.open("raw/image.jpeg")
output = model(img)
cropped_image = output.crop(save=True, save_dir=f"Result/{reference_id}")
If running the piece of code multiple time for the same image and reference id, reference_id folder keeps getting suffix incremented by 1 and hence new folders get created. However, I am looking to overwrite the very first folder. Please help