SegDecNet
SegDecNet copied to clipboard
seanXYZ
Pytorch Implement of the paper "Segmentation-Based Deep-Learning Approach for Surface Defect Detection"
error
raise ValueError("num_samples should be a positive integer " ValueError: num_samples should be a positive integer value, but got num_samples=0
Hi, Thanks for sharing! Would you mind share how you convert raw KolektorSDD to your NG/OK split dataset? I am working to reproduce all result from the paper(eg different dilation),...
I ran the code, but the accuracy was not as good as in the paper.
File "C:\Users\Administrator\anaconda3\envs\pytorch\lib\multiprocessing\spawn.py", line 138, in _check_not_importing_main raise RuntimeError(''' RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase. This probably...
I'm not able to train the model, since the loss function gives me back a Value error on the target size... do you know ho to fix it?
dataset
Hello, thanks for your share. I am confuse about why you divide the data set into trainOK and trainNG. What do NG and OK mean respectively
Can you add a licence to this repository @seanXYZ ? Would be very nice to use some of your awesome Code! :)
When running _train_segment.py_ and _train_decision.py_, the error occurs: > usage: ipykernel_launcher.py [-h] [--cuda CUDA] [--gpu_num GPU_NUM] [--worker_num WORKER_NUM] [--batch_size BATCH_SIZE] [--lr LR] [--b1 B1] [--b2 B2] [--begin_epoch BEGIN_EPOCH] [--end_epoch END_EPOCH]...
In this paper, the author said the loss function is perfectly crossentropy, but when I modify the MSELoss to CrossEntropy Loss. The code do not work. How to modify the...
I think the function of initialize_weight can be: def initialize_weights(*models): for model in models: for m in model.modules(): if isinstance(m, nn.Conv2d): nn.init.kaiming_normal_(m.weight.data, nonlinearity='relu') elif isinstance(m, nn.BatchNorm2d): m.weight.data.fill_(1.) m.bias.data.fill_(1e-4) elif isinstance(m,...