Images after inferece are overwritten

[LoaferCH]

In save_preds method:

innofw/innofw/core/datamodules/lightning_datamodules/segmentation_hdf5_dm.py

Lines 98 to 108 in f0bfa99

	def save_preds(self, preds, stage: Stages, dst_path: pathlib.Path):
	out_file_path = dst_path / "results"
	os.mkdir(out_file_path)
	for preds_batch in preds:
	for i , pred in enumerate(preds_batch):
	pred = pred.numpy()
	pred[pred < 0.3] = 0
	pred[pred > 0] = 255
	filename = out_file_path / f"out_{i}.png"
	cv2.imwrite(filename, pred[0])
	logging.info(f"Saved result to: {out_file_path}")

The value of iterator i gets reset for each batch

innofw/innofw/core/datamodules/lightning_datamodules/segmentation_hdf5_dm.py

Line 102 in f0bfa99

for i , pred in enumerate(preds_batch):

So images get same names and overwrite each batch. This i iterator should be outside the loop and get incremented after cv2.imwrite()