NEU-DET-Steel-Surface-Defect-Detection
NEU-DET-Steel-Surface-Defect-Detection copied to clipboard
siddhartamukherjee
This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.
The error message is as follows, really appreciate your help. Traceback (most recent call last): File "D:/NEU-DET-Steel-Surface-Defect-Detection-master/FPN_Resnet_Train.py", line 31, in model_trainer = Trainer(model, lr, epochs, path) File "D:\NEU-DET-Steel-Surface-Defect-Detection-master\Utilities\Trainer.py", line 54,...
It seemed the trained models aren't available in the Models directory; The pth files is found to be invalid. Can you please share the PRE-TRAINED models.
@siddhartamukherjee ,I have run your code in colab. Below is the following error, Caught FileNotFoundError in DataLoader worker process 0. Original Traceback (most recent call last): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 302,...
Hey siddhartha I absolutely loved ur work and the kind of explanation u have given for this project. I have 2 issues regarding the import for example the from Data_Extraction_Visualization...
Metadata
Owner
Metadata
This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.