MSRF-Net_PyTorch icon indicating copy to clipboard operation
MSRF-Net_PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon amlarraz

How to define number of classes ?

Open TrinhThiBaoAnh opened this issue 1 year ago • 1 comments

i saw in the train.py : num_classes = 5 Does this consider background as a class? I have a binary segmentation task, should i put num_classes = 2?

TrinhThiBaoAnh avatar Jul 22 '23 08:07 TrinhThiBaoAnh

yes! you should consider the background as an additional class

amlarraz avatar Jul 23 '23 09:07 amlarraz