|
|
@ -12,18 +12,26 @@ import ontologyWorking
|
|
|
|
|
|
|
|
|
|
|
|
url = 'http://kb.athene.tech/api/1.0/ontology/'
|
|
|
|
url = 'http://kb.athene.tech/api/1.0/ontology/'
|
|
|
|
img_path = 'data'
|
|
|
|
img_path = 'data'
|
|
|
|
img_size = (1280, 720)
|
|
|
|
img_size = (1280, 720) # Размер изображения для нормализации.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def analyse_file(uid, image_path):
|
|
|
|
def analyze_file(uid: str, image_path: str) -> None:
|
|
|
|
if ontologyWorking.is_ontology_exists(uid, url):
|
|
|
|
'''
|
|
|
|
|
|
|
|
Анализирует файл и выводит результат в консоль.
|
|
|
|
|
|
|
|
@param uid: УИД онтологии.
|
|
|
|
|
|
|
|
@param url: Базовый URL сервиса.
|
|
|
|
|
|
|
|
'''
|
|
|
|
|
|
|
|
if not ontologyWorking.is_ontology_exists(uid, url):
|
|
|
|
raise Exception(f'Онтология с uid {uid} не существует')
|
|
|
|
raise Exception(f'Онтология с uid {uid} не существует')
|
|
|
|
if not os.path.isfile(image_path):
|
|
|
|
if not os.path.isfile(image_path):
|
|
|
|
raise Exception(f'Изображение {image_path} не существует')
|
|
|
|
raise Exception(f'Изображение {image_path} не существует')
|
|
|
|
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
|
|
|
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
|
|
|
model.names = neuralNetwork.rename_entity(model.names)
|
|
|
|
model.names = neuralNetwork.rename_entity(model.names)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Распознавание изображения.
|
|
|
|
results = model(imageWorking.get_image_as_array(image_path))
|
|
|
|
results = model(imageWorking.get_image_as_array(image_path))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Создание аксиом онтологии на основе результатов распознавания.
|
|
|
|
object_properties = list()
|
|
|
|
object_properties = list()
|
|
|
|
data_properties = list()
|
|
|
|
data_properties = list()
|
|
|
|
for i, res in enumerate(results.pred):
|
|
|
|
for i, res in enumerate(results.pred):
|
|
|
@ -32,6 +40,7 @@ def analyse_file(uid, image_path):
|
|
|
|
object_properties += request[0]
|
|
|
|
object_properties += request[0]
|
|
|
|
data_properties += request[1]
|
|
|
|
data_properties += request[1]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Формирование данных для запроса к сервису работы с онтологиями.
|
|
|
|
data = {
|
|
|
|
data = {
|
|
|
|
'data':
|
|
|
|
'data':
|
|
|
|
{
|
|
|
|
{
|
|
|
@ -39,13 +48,14 @@ def analyse_file(uid, image_path):
|
|
|
|
'dataPropertyAssertions': data_properties
|
|
|
|
'dataPropertyAssertions': data_properties
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
result = {
|
|
|
|
result = {
|
|
|
|
'QueryGetNotEmpty': '',
|
|
|
|
'QueryGetNotEmpty': '',
|
|
|
|
'QueryGetCheck': '',
|
|
|
|
'QueryGetCheck': '',
|
|
|
|
'QueryGetEmpty': ''
|
|
|
|
'QueryGetEmpty': ''
|
|
|
|
}
|
|
|
|
}
|
|
|
|
params = '&'.join([f'names={query}' for query in result.keys()])
|
|
|
|
params = '&'.join([f'names={query}' for query in result.keys()])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Выполнение запроса.
|
|
|
|
response = requests.post(url + f'{uid}/query/multi?{params}', json=data).json()
|
|
|
|
response = requests.post(url + f'{uid}/query/multi?{params}', json=data).json()
|
|
|
|
if response['error']:
|
|
|
|
if response['error']:
|
|
|
|
raise Exception(response['error'])
|
|
|
|
raise Exception(response['error'])
|
|
|
@ -53,6 +63,7 @@ def analyse_file(uid, image_path):
|
|
|
|
result[query] = [resultSQWRL['name']['value'] for resultSQWRL in response['response'][query]['rows']]
|
|
|
|
result[query] = [resultSQWRL['name']['value'] for resultSQWRL in response['response'][query]['rows']]
|
|
|
|
print(f'Запрос выполнен')
|
|
|
|
print(f'Запрос выполнен')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Вывод результата.
|
|
|
|
print()
|
|
|
|
print()
|
|
|
|
print('Результат:')
|
|
|
|
print('Результат:')
|
|
|
|
if result['QueryGetNotEmpty']:
|
|
|
|
if result['QueryGetNotEmpty']:
|
|
|
@ -64,14 +75,14 @@ def analyse_file(uid, image_path):
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
print('Неизвестное состояние')
|
|
|
|
print('Неизвестное состояние')
|
|
|
|
|
|
|
|
|
|
|
|
# Вывод изображения
|
|
|
|
# Вывод изображения.
|
|
|
|
cv.imshow('result', results.render()[0][:, :, ::-1])
|
|
|
|
cv.imshow('result', results.render()[0][:, :, ::-1])
|
|
|
|
cv.waitKey(0)
|
|
|
|
cv.waitKey(0)
|
|
|
|
cv.destroyAllWindows()
|
|
|
|
cv.destroyAllWindows()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Точка входа в приложение.
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if len(sys.argv) != 3:
|
|
|
|
if len(sys.argv) != 3:
|
|
|
|
print(f'Запуск: {sys.argv[0]} <Ontology UID> <Image path>')
|
|
|
|
print(f'Запуск: {sys.argv[0]} <Ontology UID> <Image path>')
|
|
|
|
exit(1)
|
|
|
|
exit(1)
|
|
|
|
analyse_file(sys.argv[1], sys.argv[2])
|
|
|
|
analyze_file(sys.argv[1], sys.argv[2])
|
|
|
|