mantasu / glasses-detector

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

Python Script is not working #5

Closed manadopeee closed 9 months ago

manadopeee commented 11 months ago

I tried running this script

from glasses_detector import SunglassesClassifier

classifier = SunglassesClassifier(model_type="small", pretrained=True).eval()

classifier.predict( input_path="path/to/dir", output_path="path/to/output.csv", label_type="int", )

but it didn't work

so I fixed it like this.

from glasses_detector import SunglassesClassifier

classifier = SunglassesClassifier(base_model="small", pretrained=True).eval()

classifier.process( input_path="path/to/dir", output_path="path/to/output.csv", label_type="int", )

mantasu commented 11 months ago

Thanks, I'll take a look into this when reworking the package after Christmas~

mantasu commented 9 months ago

Fixed with #6