search

mantasu / glasses-detector

Glasses detection, classification and segmentation
https://mantasu.github.io/glasses-detector/
MIT License
39 stars 6 forks source link

Error load pretrained weight #1

Closed duongvanbien13081999 closed 1 year ago

duongvanbien13081999 commented 1 year ago

Hi, I'm try run predict.py to test your weight, but i got the error: "RuntimeError: Error(s) in loading state_dict for SunglassesClassifier: Missing key(s) in state_dict: "base_model.features.0.0.weight", "base_model.features.0.1.weight", ...."

mantasu commented 1 year ago

Hi, the repository is being reworked and the models are being retrained. I expect this to be ready in a couple of weeks. But I will update the instructions on how to train personal models this week.

mantasu commented 1 year ago

You may want to switch to an older commit a couple of months back to test the old weights.

mantasu commented 1 year ago

Closing this cuz just released a pip package with retrained models. Models for other tasks will also be released sometime in the future.