From f2ce33bdeb97be6bc6774197e8c9a530e89459dc Mon Sep 17 00:00:00 2001 From: Dobiasd Date: Mon, 20 Nov 2023 09:39:55 +0100 Subject: [PATCH] Change model_type to base_model in docs --- README.md | 6 +++--- docs/content/examples.rst | 6 +++--- setup.py | 6 +++--- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 3504c8d..3475b7a 100644 --- a/README.md +++ b/README.md @@ -165,7 +165,7 @@ You can import the package and its models via the python script for more flexibi ```python from glasses_detector import SunglassesClassifier -classifier = SunglassesClassifier(model_type="small", pretrained=True).eval() +classifier = SunglassesClassifier(base_model="small", pretrained=True).eval() classifier.predict( input_path="path/to/dir", @@ -179,8 +179,8 @@ Using a segmenter is similar, here is an example of using a sunglasses segmentat ```python from glasses_detector import FullSunglassesSegmenter -# model_type can also be a tuple: (classifier size, base glasses segmenter size) -segmenter = FullSunglassesSegmenter(model_type="small", pretrained=True).eval() +# base_model can also be a tuple: (classifier size, base glasses segmenter size) +segmenter = FullSunglassesSegmenter(base_model="small", pretrained=True).eval() segmenter.predict( input_path="path/to/dir", diff --git a/docs/content/examples.rst b/docs/content/examples.rst index 4566a37..33915b1 100644 --- a/docs/content/examples.rst +++ b/docs/content/examples.rst @@ -35,7 +35,7 @@ You can import the package and its models via the python script for more flexibi from glasses_detector import SunglassesClassifier - classifier = SunglassesClassifier(model_type="small", pretrained=True).eval() + classifier = SunglassesClassifier(base_model="small", pretrained=True).eval() classifier.predict( input_path="path/to/dir", @@ -50,8 +50,8 @@ Using a segmenter is similar, here is an example of using a sunglasses segmentat from glasses_detector import FullSunglassesSegmenter - # model_type can also be a tuple: (classifier size, base glasses segmenter size) - segmenter = FullSunglassesSegmenter(model_type="small", pretrained=True).eval() + # base_model can also be a tuple: (classifier size, base glasses segmenter size) + segmenter = FullSunglassesSegmenter(base_model="small", pretrained=True).eval() segmenter.predict( input_path="path/to/dir", diff --git a/setup.py b/setup.py index f62cac9..5cfe1b8 100644 --- a/setup.py +++ b/setup.py @@ -150,7 +150,7 @@ ```python from glasses_detector import SunglassesClassifier -classifier = SunglassesClassifier(model_type="small", pretrained=True).eval() +classifier = SunglassesClassifier(base_model="small", pretrained=True).eval() classifier.predict( input_path="path/to/dir", @@ -164,8 +164,8 @@ ```python from glasses_detector import FullSunglassesSegmenter -# model_type can also be a tuple: (classifier size, base glasses segmenter size) -segmenter = FullSunglassesSegmenter(model_type="small", pretrained=True).eval() +# base_model can also be a tuple: (classifier size, base glasses segmenter size) +segmenter = FullSunglassesSegmenter(base_model="small", pretrained=True).eval() segmenter.predict( input_path="path/to/dir",