Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 20:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove error log

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13891446
This commit is contained in:
mulin.lyh
2023-09-04 13:30:38 +08:00
parent de2d7cbc69
commit e78fae0135
4 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modelscope/models/cv/referring_video_object_segmentation/utils/mttr.py
View File

@@ -65,6 +65,7 @@ class MTTR(nn.Module):
# keep only the valid frames (frames which are annotated):
# (for example, in a2d-sentences only the center frame in each window is annotated).
for layer_out in backbone_out:
valid_indices = valid_indices.to(layer_out.tensors.device)
layer_out.tensors = layer_out.tensors.index_select(
0, valid_indices)
layer_out.mask = layer_out.mask.index_select(0, valid_indices)

2
modelscope/pipelines/cv/face_emotion_pipeline.py
View File

@@ -31,7 +31,7 @@ class FaceEmotionPipeline(Pipeline):
logger.info('load model done')
def preprocess(self, input: Input) -> Dict[str, Any]:
img = LoadImage.convert_to_ndarray(input['img_path'])
img = LoadImage.convert_to_ndarray(input)
return img
def forward(self, input: Dict[str, Any]) -> Dict[str, Any]:

1
modelscope/pipelines/cv/face_human_hand_detection_pipeline.py
View File

@@ -39,7 +39,6 @@ class NanoDettForFaceHumanHandDetectionPipeline(Pipeline):
cls_list, bbox_list, score_list = det_infer.inference(
self.model, self.device, input)
logger.info(cls_list, bbox_list, score_list)
return {
OutputKeys.LABELS: cls_list,
OutputKeys.BOXES: bbox_list,

2
modelscope/pipelines/cv/hand_static_pipeline.py
View File

@@ -30,7 +30,7 @@ class HandStaticPipeline(Pipeline):
logger.info('load model done')
def preprocess(self, input: Input) -> Dict[str, Any]:
img = LoadImage.convert_to_ndarray(input['img_path'])
img = LoadImage.convert_to_ndarray(input)
return img
def forward(self, input: Dict[str, Any]) -> Dict[str, Any]: