import mdtraj

traj = mdtraj.load(filename_or_filenames="output.pdb")
traj.image_molecules(inplace=True)
traj.save(filename="output.processed.pdb")