Update predict.py

predict.py

@@ -78,6 +78,7 @@ def main(predict_config: OmegaConf):
                 cur_res = np.clip(cur_res * 255, 0, 255).astype('uint8')
                 cur_res = cv2.cvtColor(cur_res, cv2.COLOR_RGB2BGR)
                 cv2.imwrite(cur_out_fname, cur_res)
+                print('wrote prediction image to :', cur_out_fname)
     except KeyboardInterrupt:
         LOGGER.warning('Interrupted by user')
     except Exception as ex: