Next to the losses and other informations Cluster2Former supports visualization of the ground truth and the predictions of the model during training and evaluation.
cfg.TB_VISUALIZATION.VIS_PERIOD
- Put visualization after that many iterationscfg.TB_VISUALIZATION.VIS_CONF_THRESHOLD
- Visualize only those predictions which has higher score the this
If you use the given apptainer container, then see the starting of the tensorboard in the container_launcher.sh
In the tensorboard on the top click to the 'IMAGE' button to see the visualizations e.g.: