diff --git a/docs/api_docs/python/face_alignment.md b/docs/api_docs/python/face_alignment.md index 85f79e2f0..9cb06ec26 100644 --- a/docs/api_docs/python/face_alignment.md +++ b/docs/api_docs/python/face_alignment.md @@ -7,3 +7,19 @@ :members: :inherited-members: ``` + +## fastdeploy.vision.facealign.FaceLandmark1000 + +```{eval-rst} +.. autoclass:: fastdeploy.vision.facealign.FaceLandmark1000 + :members: + :inherited-members: +``` + +## fastdeploy.vision.facealign.PIPNet + +```{eval-rst} +.. autoclass:: fastdeploy.vision.facealign.PIPNet + :members: + :inherited-members: +``` diff --git a/python/fastdeploy/vision/classification/contrib/resnet.py b/python/fastdeploy/vision/classification/contrib/resnet.py index 46383c159..19f200d78 100644 --- a/python/fastdeploy/vision/classification/contrib/resnet.py +++ b/python/fastdeploy/vision/classification/contrib/resnet.py @@ -24,7 +24,7 @@ class ResNet(FastDeployModel): params_file="", runtime_option=None, model_format=ModelFormat.ONNX): - """Load a image classification model exported by ResNet. + """Load a image classification model exported by torchvision.ResNet. :param model_file: (str)Path of model file, e.g resnet/resnet50.onnx :param params_file: (str)Path of parameters file, if the model_fomat is ModelFormat.ONNX, this param will be ignored, can be set as empty string