|
|
|
@ -22,7 +22,6 @@ def analyse_file(uid, image_path):
|
|
|
|
|
raise Exception(f'Изображение {image_path} не существует')
|
|
|
|
|
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
|
|
|
|
model.names = neuralNetwork.rename_entity(model.names)
|
|
|
|
|
list_sqwrl = ontologyWorking.get_list_sqwrl(uid, url)
|
|
|
|
|
|
|
|
|
|
results = model(imageWorking.get_image_as_array(image_path))
|
|
|
|
|
object_properties = list()
|
|
|
|
@ -46,15 +45,16 @@ def analyse_file(uid, image_path):
|
|
|
|
|
'QueryGetCheck': '',
|
|
|
|
|
'QueryGetEmpty': ''
|
|
|
|
|
}
|
|
|
|
|
for sqwrl in list_sqwrl['response']:
|
|
|
|
|
response = requests.post(url + f'{uid}/query/{sqwrl["name"]}', json=data).json()
|
|
|
|
|
if response['error']:
|
|
|
|
|
raise Exception(response['error'])
|
|
|
|
|
result[sqwrl['name']] = [resultSQWRL['name']['value'] for resultSQWRL in response['response']['rows']]
|
|
|
|
|
print(f'Запрос {sqwrl["name"]} выполнен')
|
|
|
|
|
params = '&'.join([f'names={query}' for query in result.keys()])
|
|
|
|
|
response = requests.post(url + f'{uid}/query/multi?{params}', json=data).json()
|
|
|
|
|
if response['error']:
|
|
|
|
|
raise Exception(response['error'])
|
|
|
|
|
for query in response['response']:
|
|
|
|
|
result[query] = [resultSQWRL['name']['value'] for resultSQWRL in response['response'][query]['rows']]
|
|
|
|
|
print(f'Запрос выполнен')
|
|
|
|
|
|
|
|
|
|
print()
|
|
|
|
|
print('Результаты')
|
|
|
|
|
print('Результат:')
|
|
|
|
|
if result['QueryGetNotEmpty']:
|
|
|
|
|
print('Аудитория занята')
|
|
|
|
|
elif result['QueryGetCheck']:
|
|
|
|
|