I want to infer the label generated by the test set

[mst272]

But there seems to be no infer part in the code, please ask how to infer

[AndreaCodegoni]

Hello,
starting from test_ondata.py you can save the masks/labels obtained from the model by saving generated_mask or bin_genmask if you want thresholded labels.