By using this callback, on every epoch change, the inference procedure can be called for 6 images in the mini_test folder. This will help you confirm how the predicted mask changes at each epoch ...

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export. The largest collection of PyTorch image encoders / backbones. Including train, eval, ...