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
get_model_output中似乎缺少了ChangeOS和SM3Det的处理分支def get_model_output(model_name: str, input_image_path: str, **args):
import pandas as pd