Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions fastdeploy/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,16 @@
# limitations under the License.
from __future__ import absolute_import
import logging
from .fastdeploy_main import Frontend, Backend, FDDataType, TensorInfo, Device
from .fastdeploy_runtime import *
from . import fastdeploy_main as C
from .fastdeploy_runtime import *
from . import vision
from .download import download, download_and_decompress

Frontend = C.Frontend
Backend = C.Backend
FDDataType = C.FDDataType
TensorInfo = C.TensorInfo
Device = C.Device

def TensorInfoStr(tensor_info):
message = "TensorInfo(name : '{}', dtype : '{}', shape : '{}')".format(
Expand Down Expand Up @@ -103,5 +107,6 @@ def RuntimeOptionStr(runtime_option):
message.strip("\n")
message += ")"
return message

C.TensorInfo.__repr__ = TensorInfoStr
C.RuntimeOption.__repr__ = RuntimeOptionStr
2 changes: 1 addition & 1 deletion fastdeploy/fastdeploy_runtime.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
# limitations under the License.
from __future__ import absolute_import
import logging
from . import fastdeploy_main as C
from . import C


class FastDeployModel:
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/deepcam/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class YOLOv5Face(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/megvii/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class YOLOX(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/meituan/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class YOLOv6(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/ppcls/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class Model(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/ppdet/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class PPYOLOE(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/ppseg/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class Model(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/rangilyu/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class NanoDetPlus(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/ultralytics/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import C


class YOLOv5(FastDeployModel):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/visualize/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from __future__ import absolute_import
import logging
from ... import fastdeploy_main as C
from ... import C


def vis_detection(im_data, det_result, line_size=1, font_size=0.5):
Expand Down
2 changes: 1 addition & 1 deletion fastdeploy/vision/wongkinyiu/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from __future__ import absolute_import
import logging
from ... import FastDeployModel, Frontend
from ... import fastdeploy_main as C
from ... import as C


class YOLOv7(FastDeployModel):
Expand Down