Giters

siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection

This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

I ran the script and got an error. How can I fix it?

Shiratsuyu1 opened this issue · comments

commented

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, in init
checkpoint = torch.load(self.PATH)
File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 593, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 762, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, 'v'.

Process finished with exit code 1

commented

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, in init checkpoint = torch.load(self.PATH) File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 593, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "C:\Users\pc.conda\envs\env\lib\site-packages\torch\serialization.py", line 762, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, 'v'.

Process finished with exit code 1

hello，I also encountered the same problem. Have you solved it？