-
Notifications
You must be signed in to change notification settings - Fork 968
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[WIP] upgrade model inference #2456
base: develop
Are you sure you want to change the base?
Conversation
Thanks for your contribution! |
1b56ed0
to
4ae6b29
Compare
777a143
to
19d0c93
Compare
93b697d
to
fe68cee
Compare
paddlex/inference/common/funcs.py
Outdated
from ..utils.color_map import get_colormap, font_colormap | ||
|
||
|
||
def create_font(txt, sz, font_path): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
paddlex/inference/common/funcs.py
Outdated
return font | ||
|
||
|
||
def draw_box_txt_fine(img_size, box, txt, font_path): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
考虑到未来有更多的common funcs,看看是不是一类操作放到一个文件里面?
raise NotImplementedError | ||
|
||
@abstractmethod | ||
def set_predictor(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
之前讨论不支持设置吧?
1c7b4db
to
fb8c28c
Compare
fb8c28c
to
1f7ddf2
Compare
def _build_batch_sampler(self): | ||
return ImageBatchSampler() | ||
|
||
def _get_result_class(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
放到transformers下面定义
# def add(self, ) | ||
|
||
|
||
class BaseComponent(ABC): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
base component和pipeline中的component是否可复用?
def _get_result_class(self): | ||
return TopkResult | ||
|
||
def _build_transformers(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
transformer中有predictor不合理,trans 需要换个名称,
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
processors
operators
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
cmpt -> component
TODO: