BiSeNet
BiSeNet copied to clipboard
Questions about training with customdatasets
I made the following changes:
the errror:
I hope someone can help me take a look, how should I modify。Thanks!!!!!!!!!
Hi,
Please make sure your input image sizes are divisible by 32.
Hi,
Please make sure your input image sizes are divisible by 32. he worked! thank you very much!! If I want to train a custom crop size, where should I modify the code so that the crop ratio is not restricted by 32.thank you!!
Excuse me, but I encountered a new problem. When I started training and reported the following error, how can I solve it? Thank you.
You can change crop size in configuration file: https://github.com/CoinCheung/BiSeNet/blob/f9231b7c971413e6ebdfcd961fbea53417b18851/configs/bisenetv1_city.py#L15
What is your dataset like? How many categories are in your dataset? What is the range of label values?
What is your dataset like? How many categories are in your dataset? What is the range of label values?
it has 3 categories ,sorry,what about the last question mean?
Please check here: https://github.com/CoinCheung/BiSeNet/blob/f9231b7c971413e6ebdfcd961fbea53417b18851/lib/cityscapes_cv2.py#L67 If your label values are within (0,1,2), you need to bypass these lines about label value mapping.
Please check here:
https://github.com/CoinCheung/BiSeNet/blob/f9231b7c971413e6ebdfcd961fbea53417b18851/lib/cityscapes_cv2.py#L67
If your label values are within (0,1,2), you need to bypass these lines about label value mapping.
Do you mean to modify this file?
I made the following changes:
My label map:
What changes should I make to solve this problem?
Have you checked every label image to make sure every label image is within [0-2] ?
There are only two categories of label images and three categories of label images in the data set
I mean, have you checked one by one, and make sure each picuture's value are within [0-2]? Better try with cv2.imread(pth, 0)
:
https://github.com/CoinCheung/BiSeNet/blob/f9231b7c971413e6ebdfcd961fbea53417b18851/lib/base_dataset.py#L53
you mean this? The output of each label map in my data set is like the following figure。
I still haven't solved this problem。
These are pytorch native operators, which is unlikely to have memory problem. What is the batch size did you use to train your model?
batchsize is 8
batchsize is 8
excuse me ! have you ever solved this problem?
batchsize is 8
i meet the same question
@Thatboy7 you should guarantee your labels have one channel.