Skip to content

get_model_output中是否缺少了ChangeOS和SM3Det的处理分支 #5

@csu-K

Description

@csu-K

get_model_output中似乎缺少了ChangeOS和SM3Det的处理分支def get_model_output(model_name: str, input_image_path: str, **args):
import pandas as pd

result = None
try:
    # classification
    if model_name in ['MSCN', 'RemoteCLIP']:
        result = results[(results['model'] == model_name) & (results['file_path'] == input_image_path)].values[0]
    # detection
    elif model_name == 'Strip-R-CNN':
        result = results[(results['model'] == model_name) & (results['file_path'] == input_image_path)].values[0]
    # visual grounding
    elif model_name == 'RemoteSAM':
        result = results[(results['model'] == model_name) & (results['file_path'] == input_image_path)].values[0]
        result = result[args['text_prompt']]
    # counting
    elif model_name == 'InstructSAM':
        result = results[(results['model'] == model_name) & (results['file_path'] == input_image_path)].values[0]
        result = result[args['text_prompt']]
    # segmentation
    elif model_name == 'SAM2':
        result = results[(results['model'] == model_name) & (results['file_path'] == input_image_path)].values[0]
        result = result[args['bbox']]
except:
    pass

if result is None:
    return 'Failed to call model'
else:
    return result

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions