Closed hitxujian closed 4 years ago
tks for your issue👍
there are 2 key parts you need to modify (without the preprocessing
part) :
dataloader
functions in /step2to4_train_validate_inference/loader/data_loader.py
, which named get_loader
and get_loader_difficult
, and you need to modify the code of the 'processing GT' part in this script (as well as the img_mask_aug.py
) to ensure that GT is loaded successfullysolver.py
to ensure that loss is calculated correctlyThese changes may not be easy to implement if you are not familiar with my code, so I strongly recommend that you familiarize yourself with the code with the demo dataset (DDTI) before making any changes, and I will consider modifying the code to accommodate multi-class segmentation in the future
thanks for your sharing! if i want to train a multi class segment,how i custom your code