Базовый коммит
This commit is contained in:
Aleksandr Mochalov
@@ -0,0 +1 @@
|
|||||||
|
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
__pycache__
|
||||||
|
.vscode
|
||||||
|
data/*
|
||||||
|
!data/demo/
|
||||||
|
!data/demo/**
|
||||||
|
client/static/*
|
||||||
|
!client/static/logo.png
|
||||||
|
!client/static/favicon.ico
|
||||||
Binary file not shown.
|
After Width: | Height: | Size: 5.6 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 88 KiB |
@@ -0,0 +1,233 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="ru">
|
||||||
|
<head>
|
||||||
|
<link rel="icon" type="image/x-icon" href="static/favicon.ico">
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Вход в сервис</title>
|
||||||
|
<style>
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
||||||
|
-webkit-font-smoothing: antialiased;
|
||||||
|
text-rendering: optimizeLegibility;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
|
||||||
|
min-height: 100vh;
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
align-items: center;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-container {
|
||||||
|
width: 100%;
|
||||||
|
max-width: 450px;
|
||||||
|
background: white;
|
||||||
|
border-radius: 15px;
|
||||||
|
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-header {
|
||||||
|
background: white;
|
||||||
|
color: #333;
|
||||||
|
padding: 25px;
|
||||||
|
text-align: center;
|
||||||
|
border-bottom: 2px solid #eee;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo-container {
|
||||||
|
background: white;
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 20px;
|
||||||
|
margin: 0 auto 20px;
|
||||||
|
display: inline-block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 200px;
|
||||||
|
height: auto;
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-header h1 {
|
||||||
|
font-size: 24px;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-header p {
|
||||||
|
color: #7f8c8d;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-content {
|
||||||
|
padding: 25px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-group {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
label {
|
||||||
|
display: block;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.flex-row {
|
||||||
|
align-items: stretch;
|
||||||
|
display: flex;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
position: relative;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--label {
|
||||||
|
height: 48px;
|
||||||
|
width: 50px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
background: #f5f6f8;
|
||||||
|
border: 2px solid #ddd;
|
||||||
|
border-right: none;
|
||||||
|
border-top-left-radius: 8px;
|
||||||
|
border-bottom-left-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--input {
|
||||||
|
flex: 1;
|
||||||
|
padding: 12px 15px;
|
||||||
|
border: 2px solid #ddd;
|
||||||
|
border-radius: 0 8px 8px 0;
|
||||||
|
font-size: 16px;
|
||||||
|
transition: border-color 0.3s;
|
||||||
|
height: 48px;
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--input:focus {
|
||||||
|
border-color: #3498db;
|
||||||
|
outline: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--submit {
|
||||||
|
display: block;
|
||||||
|
width: 100%;
|
||||||
|
padding: 15px;
|
||||||
|
background: linear-gradient(135deg, #c182b9 0%, #5ac6c8 100%);
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: opacity 0.3s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--submit:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
|
||||||
|
.error {
|
||||||
|
color: #d9534f;
|
||||||
|
background-color: #f8d7da;
|
||||||
|
border: 1px solid #f5c6cb;
|
||||||
|
padding: 15px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
border-radius: 8px;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.forgot-password {
|
||||||
|
color: #3498db;
|
||||||
|
text-decoration: none;
|
||||||
|
font-size: 14px;
|
||||||
|
transition: color 0.3s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.forgot-password:hover {
|
||||||
|
color: #2980b9;
|
||||||
|
text-decoration: underline;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 600px) {
|
||||||
|
.login-container {
|
||||||
|
max-width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 150px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.login-header h1 {
|
||||||
|
font-size: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.flex-row {
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--label {
|
||||||
|
width: 100%;
|
||||||
|
border: 2px solid #ddd;
|
||||||
|
border-bottom: none;
|
||||||
|
border-radius: 8px 8px 0 0;
|
||||||
|
padding: 5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.lf--input {
|
||||||
|
border: 2px solid #ddd;
|
||||||
|
border-top: none;
|
||||||
|
border-radius: 0 0 8px 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="login-container">
|
||||||
|
<div class="login-header">
|
||||||
|
<div class="logo-container">
|
||||||
|
<img src="static/logo.png" alt="Логотип" class="logo">
|
||||||
|
</div>
|
||||||
|
<h1>Вход в сервис</h1>
|
||||||
|
<p>Введите ваши учетные данные</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="login-content">
|
||||||
|
<!-- Сообщение об ошибке -->
|
||||||
|
<div class="error" style="display: none;">
|
||||||
|
Неправильный логин или пароль
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<form class='login-form' method="post" action="/login">
|
||||||
|
<div class="flex-row">
|
||||||
|
<label class="lf--label" for="username">
|
||||||
|
<svg x="0px" y="0px" width="12px" height="13px">
|
||||||
|
<path fill="#7f8c8d" d="M8.9,7.2C9,6.9,9,6.7,9,6.5v-4C9,1.1,7.9,0,6.5,0h-1C4.1,0,3,1.1,3,2.5v4c0,0.2,0,0.4,0.1,0.7 C1.3,7.8,0,9.5,0,11.5V13h12v-1.5C12,9.5,10.7,7.8,8.9,7.2z M4,2.5C4,1.7,4.7,1,5.5,1h1C7.3,1,8,1.7,8,2.5v4c0,0.2,0,0.4-0.1,0.6 l0.1,0L7.9,7.3C7.6,7.8,7.1,8.2,6.5,8.2h-1c-0.6,0-1.1-0.4-1.4-0.9L4.1,7.1l0.1,0C4,6.9,4,6.7,4,6.5V2.5z M11,12H1v-0.5 c0-1.6,1-2.9,2.4-3.4c0.5,0.7,1.2,1.1,2.1,1.1h1c0.8,0,1.6-0.4,2.1-1.1C10,8.5,11,9.9,11,11.5V12z"/>
|
||||||
|
</svg>
|
||||||
|
</label>
|
||||||
|
<input id="username" name="username" class='lf--input' placeholder='Логин' type='text' required>
|
||||||
|
</div>
|
||||||
|
<div class="flex-row">
|
||||||
|
<label class="lf--label" for="password">
|
||||||
|
<svg x="0px" y="0px" width="15px" height="5px">
|
||||||
|
<g>
|
||||||
|
<path fill="#7f8c8d" d="M6,2L6,2c0-1.1-1-2-2.1-2H2.1C1,0,0,0.9,0,2.1v0.8C0,4.1,1,5,2.1,5h1.7C5,5,6,4.1,6,2.9V3h5v1h1V3h1v2h1V3h1 V2H6z M5.1,2.9c0,0.7-0.6,1.2-1.3,1.2H2.1c-0.7,0-1.3-0.6-1.3-1.2V2.1c0-0.7,0.6-1.2,1.3-1.2h1.7c0.7,0,1.3,0.6,1.3,1.2V2.9z"/>
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
</label>
|
||||||
|
<input id="password" name="password" class='lf--input' placeholder='Пароль' type='password' required>
|
||||||
|
</div>
|
||||||
|
<input class='lf--submit' type='submit' value='ВОЙТИ'>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@@ -0,0 +1,294 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="ru">
|
||||||
|
<head>
|
||||||
|
<link rel="icon" type="image/x-icon" href="static/favicon.ico">
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Результаты анализа DICOM</title>
|
||||||
|
<style>
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
|
||||||
|
min-height: 100vh;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.container {
|
||||||
|
max-width: 1200px;
|
||||||
|
margin: 0 auto;
|
||||||
|
background: white;
|
||||||
|
border-radius: 15px;
|
||||||
|
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header {
|
||||||
|
background: white;
|
||||||
|
color: #333;
|
||||||
|
padding: 25px;
|
||||||
|
text-align: center;
|
||||||
|
border-bottom: 2px solid #eee;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo-container {
|
||||||
|
background: white;
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 20px;
|
||||||
|
margin: 0 auto 20px;
|
||||||
|
display: inline-block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 200px;
|
||||||
|
height: auto;
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header h1 {
|
||||||
|
font-size: 24px;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header p {
|
||||||
|
color: #7f8c8d;
|
||||||
|
}
|
||||||
|
|
||||||
|
.content {
|
||||||
|
padding: 25px;
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: 1fr 1fr;
|
||||||
|
gap: 30px;
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 900px) {
|
||||||
|
.content {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 150px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.result-card {
|
||||||
|
background: #f8f9fa;
|
||||||
|
border-radius: 10px;
|
||||||
|
padding: 20px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.result-card h3 {
|
||||||
|
color: #2c3e50;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
padding-bottom: 10px;
|
||||||
|
border-bottom: 2px solid #5ac6c8;
|
||||||
|
}
|
||||||
|
|
||||||
|
.probability-display {
|
||||||
|
text-align: center;
|
||||||
|
padding: 20px;
|
||||||
|
background: white;
|
||||||
|
border-radius: 10px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
box-shadow: 0 4px 8px rgba(0,0,0,0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.probability-value {
|
||||||
|
font-size: 48px;
|
||||||
|
font-weight: bold;
|
||||||
|
color: #2c3e50;
|
||||||
|
margin: 10px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.probability-label {
|
||||||
|
font-size: 18px;
|
||||||
|
color: #7f8c8d;
|
||||||
|
}
|
||||||
|
|
||||||
|
.pathology-badge {
|
||||||
|
display: inline-block;
|
||||||
|
padding: 8px 16px;
|
||||||
|
border-radius: 20px;
|
||||||
|
font-weight: 600;
|
||||||
|
margin-top: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.pathology-yes {
|
||||||
|
background: linear-gradient(135deg, #c182b9 0%, #e74c3c 100%);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.pathology-no {
|
||||||
|
background: linear-gradient(135deg, #5ac6c8 0%, #2ecc71 100%);
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
.report-section {
|
||||||
|
background: white;
|
||||||
|
padding: 20px;
|
||||||
|
border-radius: 10px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.report-section h3 {
|
||||||
|
color: #2c3e50;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.report-content {
|
||||||
|
line-height: 1.6;
|
||||||
|
color: #34495e;
|
||||||
|
}
|
||||||
|
|
||||||
|
.conclusion {
|
||||||
|
background: #e8fcfb;
|
||||||
|
padding: 20px;
|
||||||
|
border-radius: 10px;
|
||||||
|
border-left: 4px solid #5ac6c8;
|
||||||
|
margin: 20px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.image-container {
|
||||||
|
text-align: center;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dicom-image {
|
||||||
|
max-width: 100%;
|
||||||
|
border-radius: 8px;
|
||||||
|
box-shadow: 0 4px 8px rgba(0,0,0,0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.properties-table {
|
||||||
|
width: 100%;
|
||||||
|
border-collapse: collapse;
|
||||||
|
margin: 15px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.properties-table th, .properties-table td {
|
||||||
|
padding: 12px;
|
||||||
|
text-align: left;
|
||||||
|
border-bottom: 1px solid #ddd;
|
||||||
|
}
|
||||||
|
|
||||||
|
.properties-table th {
|
||||||
|
background-color: #f2f2f2;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.actions {
|
||||||
|
display: flex;
|
||||||
|
gap: 15px;
|
||||||
|
margin-top: 30px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn {
|
||||||
|
padding: 12px 25px;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: opacity 0.3s;
|
||||||
|
text-decoration: none;
|
||||||
|
display: inline-block;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary {
|
||||||
|
background: linear-gradient(135deg, #c182b9 0%, #5ac6c8 100%);
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary {
|
||||||
|
background-color: #f8f9fa;
|
||||||
|
color: #2c3e50;
|
||||||
|
border: 1px solid #ddd;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-secondary:hover {
|
||||||
|
background-color: #e9ecef;
|
||||||
|
}
|
||||||
|
|
||||||
|
.footer {
|
||||||
|
text-align: center;
|
||||||
|
padding: 20px;
|
||||||
|
background: #2c3e50;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="container">
|
||||||
|
<div class="header">
|
||||||
|
<div class="logo-container">
|
||||||
|
<img src="static/logo.png" alt="Логотип" class="logo">
|
||||||
|
</div>
|
||||||
|
<h1>Результаты анализа DICOM файла</h1>
|
||||||
|
<p>Анализ завершен. Ниже представлены результаты.</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="content">
|
||||||
|
<div>
|
||||||
|
<div class="probability-display">
|
||||||
|
<div class="probability-label">Вероятность патологии</div>
|
||||||
|
<div class="probability-value">{{ prediction.overall_probability }}%</div>
|
||||||
|
<div class="pathology-badge {{ 'pathology-yes' if prediction.is_pathology else 'pathology-no' }}">
|
||||||
|
{{ 'Обнаружена патология' if prediction.is_pathology else 'Патология не обнаружена' }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="image-container">
|
||||||
|
<img src="static/{{ prediction.image }}" alt="Результат" class="dicom-image">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
<div class="report-section">
|
||||||
|
<h3>Отчет анализа</h3>
|
||||||
|
<div class="report-content">
|
||||||
|
{{ prediction.report }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="conclusion">
|
||||||
|
<h3>Заключение</h3>
|
||||||
|
<p>{{ prediction.conclusion }}</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="result-card">
|
||||||
|
<h3>Доп. информация</h3>
|
||||||
|
<table class="properties-table">
|
||||||
|
{% for key, value in prediction.properties.items() %}
|
||||||
|
<tr>
|
||||||
|
<th>{{ key }}</th>
|
||||||
|
<td>{{ value }}</td>
|
||||||
|
</tr>
|
||||||
|
{% endfor %}
|
||||||
|
</table>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="actions">
|
||||||
|
<a href="/upload-study" class="btn btn-primary">Новый анализ</a>
|
||||||
|
<a href="/download/{{prediction.image.replace('.png', '.zip')}}" class="btn btn-secondary">Сохранить DCM отчеты</a>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="footer">
|
||||||
|
<p>Результаты анализа сгенерированы автоматически. В исследовательских целях.</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@@ -0,0 +1,615 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="ru">
|
||||||
|
<head>
|
||||||
|
<link rel="icon" type="image/x-icon" href="static/favicon.ico">
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Анализ DICOM файлов</title>
|
||||||
|
<style>
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
|
||||||
|
min-height: 100vh;
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
align-items: center;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.container {
|
||||||
|
width: 100%;
|
||||||
|
max-width: 800px;
|
||||||
|
background: white;
|
||||||
|
border-radius: 15px;
|
||||||
|
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
|
||||||
|
overflow: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header {
|
||||||
|
background: white;
|
||||||
|
color: #333;
|
||||||
|
padding: 25px;
|
||||||
|
text-align: center;
|
||||||
|
border-bottom: 2px solid #eee;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo-container {
|
||||||
|
background: white;
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 20px;
|
||||||
|
margin: 0 auto 20px;
|
||||||
|
display: inline-block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 200px;
|
||||||
|
height: auto;
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header h1 {
|
||||||
|
font-size: 24px;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header p {
|
||||||
|
color: #7f8c8d;
|
||||||
|
}
|
||||||
|
|
||||||
|
.content {
|
||||||
|
padding: 25px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-group {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
label {
|
||||||
|
display: block;
|
||||||
|
margin-bottom: 8px;
|
||||||
|
font-weight: 600;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
width: 100%;
|
||||||
|
padding: 12px 15px;
|
||||||
|
border: 2px solid #ddd;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 16px;
|
||||||
|
transition: border-color 0.3s;
|
||||||
|
}
|
||||||
|
|
||||||
|
select:focus {
|
||||||
|
border-color: #5ac6c8;
|
||||||
|
outline: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.demo-files {
|
||||||
|
background-color: #f8f9fa;
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 15px;
|
||||||
|
margin-top: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.demo-files h3 {
|
||||||
|
margin-bottom: 12px;
|
||||||
|
color: #2c3e50;
|
||||||
|
font-size: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.file-list {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
|
||||||
|
gap: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.demo-file {
|
||||||
|
background: white;
|
||||||
|
border: 1px solid #e0e0e0;
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 12px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.3s;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.demo-file:hover {
|
||||||
|
border-color: #5ac6c8;
|
||||||
|
box-shadow: 0 5px 15px rgba(90, 198, 200, 0.2);
|
||||||
|
transform: translateY(-2px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.demo-file.selected {
|
||||||
|
border-color: #5ac6c8;
|
||||||
|
background-color: #e8fcfb;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone {
|
||||||
|
border: 2px dashed #5ac6c8;
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 40px 20px;
|
||||||
|
text-align: center;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.3s;
|
||||||
|
background-color: #f8f9fa;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone.highlight {
|
||||||
|
background-color: #e8fcfb;
|
||||||
|
border-color: #c182b9;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone p {
|
||||||
|
margin-bottom: 15px;
|
||||||
|
color: #7f8c8d;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drop-zone .icon {
|
||||||
|
font-size: 48px;
|
||||||
|
color: #5ac6c8;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn {
|
||||||
|
display: block;
|
||||||
|
width: 100%;
|
||||||
|
padding: 15px;
|
||||||
|
background: linear-gradient(135deg, #c182b9 0%, #5ac6c8 100%);
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
border-radius: 8px;
|
||||||
|
font-size: 16px;
|
||||||
|
font-weight: 600;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: opacity 0.3s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn:hover {
|
||||||
|
opacity: 0.9;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn:disabled {
|
||||||
|
background: #bdc3c7;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
|
||||||
|
.file-info {
|
||||||
|
margin-top: 20px;
|
||||||
|
padding: 15px;
|
||||||
|
background-color: #f8f9fa;
|
||||||
|
border-radius: 8px;
|
||||||
|
border-left: 4px solid #5ac6c8;
|
||||||
|
}
|
||||||
|
|
||||||
|
.file-info h3 {
|
||||||
|
margin-bottom: 10px;
|
||||||
|
color: #2c3e50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.file-details {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
margin-top: 10px;
|
||||||
|
padding-top: 10px;
|
||||||
|
border-top: 1px solid #e0e0e0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification {
|
||||||
|
padding: 15px;
|
||||||
|
margin: 20px 0;
|
||||||
|
border-radius: 8px;
|
||||||
|
text-align: center;
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification.success {
|
||||||
|
background-color: #d4edda;
|
||||||
|
color: #155724;
|
||||||
|
border: 1px solid #c3e6cb;
|
||||||
|
display: block !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.notification.error {
|
||||||
|
background-color: #f8d7da;
|
||||||
|
color: #721c24;
|
||||||
|
border: 1px solid #f5c6cb;
|
||||||
|
display: block !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.upload-section {
|
||||||
|
margin-top: 30px;
|
||||||
|
padding-top: 20px;
|
||||||
|
border-top: 1px solid #eee;
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-title {
|
||||||
|
font-size: 18px;
|
||||||
|
color: #2c3e50;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.section-title::before {
|
||||||
|
content: "•";
|
||||||
|
margin-right: 10px;
|
||||||
|
color: #5ac6c8;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading {
|
||||||
|
display: inline-block;
|
||||||
|
width: 20px;
|
||||||
|
height: 20px;
|
||||||
|
border: 3px solid rgba(255,255,255,.3);
|
||||||
|
border-radius: 50%;
|
||||||
|
border-top-color: #fff;
|
||||||
|
animation: spin 1s ease-in-out infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes spin {
|
||||||
|
to { transform: rotate(360deg); }
|
||||||
|
}
|
||||||
|
|
||||||
|
@media (max-width: 600px) {
|
||||||
|
.file-list {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header h1 {
|
||||||
|
font-size: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.logo {
|
||||||
|
max-width: 150px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="container">
|
||||||
|
<div class="header">
|
||||||
|
<div class="logo-container">
|
||||||
|
<img src="static/logo.png" alt="Логотип" class="logo">
|
||||||
|
</div>
|
||||||
|
<h1>Выявление патологий в лучевых исследованиях</h1>
|
||||||
|
<p>Загрузите DICOM файл или выберите демонстрационный пример</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="content">
|
||||||
|
{% if error %}
|
||||||
|
<div class="notification error">
|
||||||
|
{{ error }}
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<form id="upload-form" action="/upload-study" method="post" enctype="multipart/form-data">
|
||||||
|
<div class="form-group">
|
||||||
|
<label for="pathology">Выберите искомую патологию:</label>
|
||||||
|
<select id="pathology" name="pathology">
|
||||||
|
<option value="sinus">Синусит</option>
|
||||||
|
<option value="wrist">Перелом костей лучезапястного сустава</option>
|
||||||
|
<option value="shoulder">Перелом костей плечевого сустава</option>
|
||||||
|
</select>
|
||||||
|
|
||||||
|
<div class="demo-files">
|
||||||
|
<h3>Демонстрационные файлы:</h3>
|
||||||
|
<div class="file-list" id="demo-files-list">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="upload-section">
|
||||||
|
<div class="section-title">Или загрузите свой файл</div>
|
||||||
|
|
||||||
|
<div class="drop-zone" id="drop-zone">
|
||||||
|
<div class="icon">📁</div>
|
||||||
|
<p>Перетащите DICOM файл сюда или нажмите для выбора</p>
|
||||||
|
<small>Поддерживаются только файлы в формате .dcm</small>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<input type="file" id="file-input" name="file" accept=".dcm" style="display: none;">
|
||||||
|
|
||||||
|
<input type="hidden" id="demo-filename" name="demo_filename" value="">
|
||||||
|
|
||||||
|
<div id="notification" class="notification"></div>
|
||||||
|
|
||||||
|
<button type="submit" id="analyze-btn" class="btn" disabled>
|
||||||
|
<span id="btn-text">Запустить анализ</span>
|
||||||
|
<span id="btn-loading" class="loading" style="display: none;"></span>
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<div id="file-info" class="file-info" style="display: none;">
|
||||||
|
<h3>Информация о файле</h3>
|
||||||
|
<div id="file-details" class="file-details">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
document.addEventListener('DOMContentLoaded', function() {
|
||||||
|
const form = document.getElementById('upload-form');
|
||||||
|
const pathologySelect = document.getElementById('pathology');
|
||||||
|
const demoFilesList = document.getElementById('demo-files-list');
|
||||||
|
const dropZone = document.getElementById('drop-zone');
|
||||||
|
const fileInput = document.getElementById('file-input');
|
||||||
|
const demoFilenameInput = document.getElementById('demo-filename');
|
||||||
|
const analyzeBtn = document.getElementById('analyze-btn');
|
||||||
|
const btnText = document.getElementById('btn-text');
|
||||||
|
const btnLoading = document.getElementById('btn-loading');
|
||||||
|
const fileInfo = document.getElementById('file-info');
|
||||||
|
const fileDetails = document.getElementById('file-details');
|
||||||
|
const notification = document.getElementById('notification');
|
||||||
|
|
||||||
|
let selectedFile = null;
|
||||||
|
let selectedDemoFile = null;
|
||||||
|
|
||||||
|
const demoFiles = {
|
||||||
|
sinus: [
|
||||||
|
{ name: "Рентген пазух носа (норма)", filename: "sinus_normal.dcm" },
|
||||||
|
{ name: "Рентген пазух носа (синусит)", filename: "sinusitis.dcm" }
|
||||||
|
],
|
||||||
|
wrist: [
|
||||||
|
{ name: "Рентген запястья (норма)", filename: "wrist_normal.dcm" },
|
||||||
|
{ name: "Рентген запястья (перелом)", filename: "wrist_fracture.dcm" }
|
||||||
|
],
|
||||||
|
shoulder: [
|
||||||
|
{ name: "Рентген плеча (норма)", filename: "shoulder_normal.dcm" },
|
||||||
|
{ name: "Рентген плеча (перелом)", filename: "shoulder_fracture.dcm" }
|
||||||
|
]
|
||||||
|
};
|
||||||
|
|
||||||
|
function showDemoFiles() {
|
||||||
|
const pathology = pathologySelect.value;
|
||||||
|
const files = demoFiles[pathology];
|
||||||
|
|
||||||
|
demoFilesList.innerHTML = '';
|
||||||
|
|
||||||
|
files.forEach(file => {
|
||||||
|
const fileElement = document.createElement('div');
|
||||||
|
fileElement.className = 'demo-file';
|
||||||
|
fileElement.textContent = file.name;
|
||||||
|
fileElement.dataset.filename = file.filename;
|
||||||
|
|
||||||
|
fileElement.addEventListener('click', function() {
|
||||||
|
document.querySelectorAll('.demo-file').forEach(f => {
|
||||||
|
f.classList.remove('selected');
|
||||||
|
});
|
||||||
|
|
||||||
|
this.classList.add('selected');
|
||||||
|
|
||||||
|
fileInput.value = '';
|
||||||
|
selectedFile = null;
|
||||||
|
|
||||||
|
selectedDemoFile = {
|
||||||
|
name: file.name,
|
||||||
|
filename: file.filename,
|
||||||
|
pathology: pathology
|
||||||
|
};
|
||||||
|
|
||||||
|
demoFilenameInput.value = file.filename;
|
||||||
|
|
||||||
|
showFileInfo(file.name, 'Демонстрационный файл');
|
||||||
|
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
|
||||||
|
showNotification('Демонстрационный файл выбран. Нажмите "Запустить анализ" для обработки.', 'success');
|
||||||
|
});
|
||||||
|
|
||||||
|
demoFilesList.appendChild(fileElement);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function showFileInfo(name, info) {
|
||||||
|
fileInfo.style.display = 'block';
|
||||||
|
fileDetails.innerHTML = `
|
||||||
|
<div>
|
||||||
|
<strong>Имя файла:</strong> ${name}
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<strong>Тип:</strong> ${info}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
function showNotification(message, type) {
|
||||||
|
notification.textContent = message;
|
||||||
|
notification.className = `notification ${type}`;
|
||||||
|
notification.style.display = 'block';
|
||||||
|
}
|
||||||
|
|
||||||
|
function hideNotification() {
|
||||||
|
notification.style.display = 'none';
|
||||||
|
notification.className = 'notification';
|
||||||
|
notification.textContent = '';
|
||||||
|
}
|
||||||
|
|
||||||
|
function restoreDemoFileState() {
|
||||||
|
const demoFilename = demoFilenameInput.value;
|
||||||
|
if (demoFilename) {
|
||||||
|
const pathology = pathologySelect.value;
|
||||||
|
const files = demoFiles[pathology];
|
||||||
|
const file = files.find(f => f.filename === demoFilename);
|
||||||
|
|
||||||
|
if (file) {
|
||||||
|
selectedDemoFile = {
|
||||||
|
name: file.name,
|
||||||
|
filename: file.filename,
|
||||||
|
pathology: pathology
|
||||||
|
};
|
||||||
|
|
||||||
|
document.querySelectorAll('.demo-file').forEach(f => {
|
||||||
|
f.classList.remove('selected');
|
||||||
|
if (f.dataset.filename === demoFilename) {
|
||||||
|
f.classList.add('selected');
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
showFileInfo(file.name, 'Демонстрационный файл');
|
||||||
|
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
|
||||||
|
showNotification('Демонстрационный файл выбран. Нажмите "Запустить анализ" для обработки.', 'success');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatFileSize(bytes) {
|
||||||
|
if (!bytes || bytes === 0) return 'Неизвестно';
|
||||||
|
const k = 1024;
|
||||||
|
const sizes = ['Bytes', 'KB', 'MB', 'GB'];
|
||||||
|
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
||||||
|
return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
pathologySelect.addEventListener('change', function() {
|
||||||
|
showDemoFiles();
|
||||||
|
|
||||||
|
selectedFile = null;
|
||||||
|
selectedDemoFile = null;
|
||||||
|
demoFilenameInput.value = '';
|
||||||
|
fileInfo.style.display = 'none';
|
||||||
|
analyzeBtn.disabled = true;
|
||||||
|
|
||||||
|
fileInput.value = '';
|
||||||
|
|
||||||
|
document.querySelectorAll('.demo-file').forEach(f => {
|
||||||
|
f.classList.remove('selected');
|
||||||
|
});
|
||||||
|
|
||||||
|
hideNotification();
|
||||||
|
});
|
||||||
|
|
||||||
|
dropZone.addEventListener('click', () => {
|
||||||
|
fileInput.click();
|
||||||
|
});
|
||||||
|
|
||||||
|
dropZone.addEventListener('dragover', (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropZone.classList.add('highlight');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropZone.addEventListener('dragleave', () => {
|
||||||
|
dropZone.classList.remove('highlight');
|
||||||
|
});
|
||||||
|
|
||||||
|
dropZone.addEventListener('drop', (e) => {
|
||||||
|
e.preventDefault();
|
||||||
|
dropZone.classList.remove('highlight');
|
||||||
|
|
||||||
|
if (e.dataTransfer.files.length) {
|
||||||
|
handleFileSelect(e.dataTransfer.files[0]);
|
||||||
|
|
||||||
|
document.querySelectorAll('.demo-file').forEach(f => {
|
||||||
|
f.classList.remove('selected');
|
||||||
|
});
|
||||||
|
selectedDemoFile = null;
|
||||||
|
demoFilenameInput.value = '';
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
fileInput.addEventListener('change', () => {
|
||||||
|
if (fileInput.files.length) {
|
||||||
|
handleFileSelect(fileInput.files[0]);
|
||||||
|
|
||||||
|
document.querySelectorAll('.demo-file').forEach(f => {
|
||||||
|
f.classList.remove('selected');
|
||||||
|
});
|
||||||
|
selectedDemoFile = null;
|
||||||
|
demoFilenameInput.value = '';
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
function handleFileSelect(file) {
|
||||||
|
if (file.name.endsWith('.dcm')) {
|
||||||
|
selectedFile = file;
|
||||||
|
selectedDemoFile = null;
|
||||||
|
demoFilenameInput.value = '';
|
||||||
|
showFileInfo(file.name, formatFileSize(file.size));
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
showNotification('Файл успешно загружен. Нажмите "Запустить анализ" для обработки.', 'success');
|
||||||
|
} else {
|
||||||
|
showNotification('Пожалуйста, выберите файл в формате DICOM (.dcm).', 'error');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
form.addEventListener('submit', function(e) {
|
||||||
|
e.preventDefault();
|
||||||
|
|
||||||
|
btnText.style.display = 'none';
|
||||||
|
btnLoading.style.display = 'inline-block';
|
||||||
|
analyzeBtn.disabled = true;
|
||||||
|
hideNotification();
|
||||||
|
|
||||||
|
const formData = new FormData();
|
||||||
|
formData.append('pathology', pathologySelect.value);
|
||||||
|
|
||||||
|
if (selectedDemoFile) {
|
||||||
|
formData.append('demo_filename', selectedDemoFile.filename);
|
||||||
|
} else if (selectedFile) {
|
||||||
|
formData.append('file', selectedFile);
|
||||||
|
} else {
|
||||||
|
showNotification('Пожалуйста, выберите файл или демонстрационный пример.', 'error');
|
||||||
|
btnText.style.display = 'inline';
|
||||||
|
btnLoading.style.display = 'none';
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
fetch('/upload-study', {
|
||||||
|
method: 'POST',
|
||||||
|
body: formData,
|
||||||
|
headers: {
|
||||||
|
'X-Requested-With': 'XMLHttpRequest'
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.then(response => {
|
||||||
|
if (response.ok) {
|
||||||
|
return response.json();
|
||||||
|
} else {
|
||||||
|
return response.json().then(err => { throw err; });
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.then(data => {
|
||||||
|
if (data.redirect) {
|
||||||
|
window.location.href = data.redirect;
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.catch(error => {
|
||||||
|
console.error('Ошибка:', error);
|
||||||
|
const message = error.error || 'Произошла неизвестная ошибка.';
|
||||||
|
showNotification(message, 'error');
|
||||||
|
btnText.style.display = 'inline';
|
||||||
|
btnLoading.style.display = 'none';
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
window.addEventListener('pageshow', function(event) {
|
||||||
|
btnText.style.display = 'inline';
|
||||||
|
btnLoading.style.display = 'none';
|
||||||
|
analyzeBtn.disabled = false;
|
||||||
|
|
||||||
|
if (notification.textContent === 'Файл отправляется на анализ...') {
|
||||||
|
hideNotification();
|
||||||
|
}
|
||||||
|
|
||||||
|
restoreDemoFileState();
|
||||||
|
});
|
||||||
|
|
||||||
|
showDemoFiles();
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,130 @@
|
|||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import uuid
|
||||||
|
from typing import Annotated
|
||||||
|
from fastapi import FastAPI, Request, UploadFile, Depends, Form, File, Cookie
|
||||||
|
from fastapi.templating import Jinja2Templates
|
||||||
|
from fastapi.security import OAuth2PasswordRequestForm
|
||||||
|
from fastapi.responses import HTMLResponse, JSONResponse, RedirectResponse, FileResponse
|
||||||
|
from fastapi.staticfiles import StaticFiles
|
||||||
|
from starlette.middleware.sessions import SessionMiddleware
|
||||||
|
from service import auth, reports
|
||||||
|
from service.db_requests import get_user
|
||||||
|
from service import structs
|
||||||
|
from service.models import User
|
||||||
|
|
||||||
|
app = FastAPI()
|
||||||
|
templates = Jinja2Templates(directory="client/templates")
|
||||||
|
app.mount("/static", StaticFiles(directory="client/static"), name="static")
|
||||||
|
|
||||||
|
app.add_middleware(
|
||||||
|
SessionMiddleware,
|
||||||
|
secret_key="CHANGEME",
|
||||||
|
session_cookie="session"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/", response_class=RedirectResponse)
|
||||||
|
async def main_page(access_token: Annotated[str | None, Cookie()] = None):
|
||||||
|
if access_token:
|
||||||
|
try:
|
||||||
|
await auth.get_current_user(access_token)
|
||||||
|
except:
|
||||||
|
return RedirectResponse("/login")
|
||||||
|
return RedirectResponse("/upload-study")
|
||||||
|
else:
|
||||||
|
return RedirectResponse("/login")
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/login", response_class=HTMLResponse)
|
||||||
|
async def login_page(request: Request):
|
||||||
|
access_token = request.cookies.get("access_token")
|
||||||
|
if access_token:
|
||||||
|
try:
|
||||||
|
await auth.get_current_user(access_token)
|
||||||
|
return RedirectResponse("/upload-study", status_code=303)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
response = templates.TemplateResponse("login.html", {"request": request})
|
||||||
|
response.headers["Cache-Control"] = "no-store, no-cache, must-revalidate, private"
|
||||||
|
response.headers["Pragma"] = "no-cache"
|
||||||
|
response.headers["Expires"] = "0"
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@app.post("/login")
|
||||||
|
async def login_submit(request: Request,
|
||||||
|
login_data: OAuth2PasswordRequestForm = Depends()):
|
||||||
|
user = get_user(login_data.username)
|
||||||
|
if user is None or not auth.verify_password(login_data.password,
|
||||||
|
user.hashed_password):
|
||||||
|
return templates.TemplateResponse("login.html",
|
||||||
|
{"request": request, "error": True})
|
||||||
|
response = RedirectResponse("/upload-study", status_code=303)
|
||||||
|
response.set_cookie(
|
||||||
|
key = "access_token",
|
||||||
|
value = auth.create_access_token(user.username),
|
||||||
|
httponly=True
|
||||||
|
)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/upload-study", dependencies=[Depends(auth.get_current_user)])
|
||||||
|
async def upload_page(request: Request):
|
||||||
|
return templates.TemplateResponse("upload.html", {"request": request})
|
||||||
|
|
||||||
|
|
||||||
|
@app.post("/upload-study")
|
||||||
|
async def study_submit(
|
||||||
|
request: Request,
|
||||||
|
user: User = Depends(auth.get_current_user),
|
||||||
|
file: UploadFile = File(None),
|
||||||
|
demo_filename: str = Form(None),
|
||||||
|
pathology: str = Form(...)
|
||||||
|
):
|
||||||
|
user_dir = os.path.join("data", user.username)
|
||||||
|
os.makedirs(user_dir, exist_ok=True)
|
||||||
|
fpath = os.path.join(user_dir, str(uuid.uuid4()) + ".dcm")
|
||||||
|
|
||||||
|
if file and file.filename:
|
||||||
|
with open(fpath, 'wb') as buffer:
|
||||||
|
buffer.write(await file.read())
|
||||||
|
else:
|
||||||
|
demo_source_path = os.path.join("data/demo", pathology, demo_filename)
|
||||||
|
shutil.copy2(demo_source_path, fpath)
|
||||||
|
|
||||||
|
try:
|
||||||
|
prediction = reports.make_reports(pathology, fpath, user.username)
|
||||||
|
except (structs.TagError, structs.ImagesError) as e:
|
||||||
|
error_msg = e.msg + ". Загрузите другой файл."
|
||||||
|
if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
|
||||||
|
return JSONResponse({"error": error_msg}, status_code=400)
|
||||||
|
else:
|
||||||
|
return templates.TemplateResponse("upload.html", {"request": request, "error": error_msg})
|
||||||
|
except Exception as e:
|
||||||
|
error_msg = "Ошибка анализа. Загрузите другой файл."
|
||||||
|
if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
|
||||||
|
return JSONResponse({"error": error_msg}, status_code=500)
|
||||||
|
else:
|
||||||
|
return templates.TemplateResponse("upload.html", {"request": request, "error": error_msg})
|
||||||
|
|
||||||
|
request.session["prediction"] = prediction._asdict()
|
||||||
|
if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
|
||||||
|
return JSONResponse({"redirect": "/ai-result"})
|
||||||
|
else:
|
||||||
|
return RedirectResponse("/ai-result", status_code=303)
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/ai-result", dependencies=[Depends(auth.get_current_user)])
|
||||||
|
async def result_page(request: Request):
|
||||||
|
prediction = request.session["prediction"]
|
||||||
|
return templates.TemplateResponse("result.html", {
|
||||||
|
"request": request,
|
||||||
|
"prediction": prediction
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@app.get("/download/{filename}")
|
||||||
|
async def download_report(filename: str,
|
||||||
|
user: User = Depends(auth.get_current_user)):
|
||||||
|
return FileResponse(f"data/{user.username}/reports/{filename}")
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
pydicom==3.0.1
|
||||||
|
pylibjpeg==2.0.1
|
||||||
|
pylibjpeg-libjpeg==2.3.0
|
||||||
|
numpy==2.2.3
|
||||||
|
opencv-python==4.11.0.86
|
||||||
|
scikit-image==0.25.2
|
||||||
|
python-multipart==0.0.20
|
||||||
|
fastapi==0.115.12
|
||||||
|
uvicorn[standard]==0.34.3
|
||||||
|
python-jose[cryptography]==3.5.0
|
||||||
|
passlib[bcrypt]==1.7.4
|
||||||
|
sqlalchemy==2.0.41
|
||||||
|
psycopg2-binary==2.9.10
|
||||||
|
itsdangerous==2.2.0
|
||||||
|
segmentation-models-pytorch==0.4.0
|
||||||
|
--extra-index-url https://download.pytorch.org/whl/cu118
|
||||||
|
torch==2.6.0+cu118
|
||||||
|
torchvision==0.21.0+cu118
|
||||||
@@ -0,0 +1,70 @@
|
|||||||
|
from typing import Annotated
|
||||||
|
from datetime import datetime, timezone
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
from fastapi import Depends, HTTPException, status, Cookie
|
||||||
|
from fastapi.security import OAuth2PasswordBearer
|
||||||
|
from passlib.context import CryptContext
|
||||||
|
from jose import jwt
|
||||||
|
from jose.exceptions import JWTError
|
||||||
|
from pydantic import ValidationError
|
||||||
|
|
||||||
|
from service.schemas import TokenData, UserData
|
||||||
|
from service.db_requests import get_user
|
||||||
|
|
||||||
|
|
||||||
|
reuseable_oauth = OAuth2PasswordBearer(
|
||||||
|
tokenUrl="login",
|
||||||
|
scheme_name="JWT"
|
||||||
|
)
|
||||||
|
|
||||||
|
ACCESS_TOKEN_EXPIRE_MINUTES = 60 * 24 * 7
|
||||||
|
ALGORITHM = "HS256"
|
||||||
|
JWT_SECRET_KEY = "CHAGEME"
|
||||||
|
|
||||||
|
password_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
|
||||||
|
|
||||||
|
|
||||||
|
def hash_password(password):
|
||||||
|
return password_context.hash(password)
|
||||||
|
|
||||||
|
|
||||||
|
def verify_password(password, hashed_pass):
|
||||||
|
return password_context.verify(password, hashed_pass)
|
||||||
|
|
||||||
|
|
||||||
|
def create_access_token(subject):
|
||||||
|
expires_delta = datetime.now() + timedelta(minutes=
|
||||||
|
ACCESS_TOKEN_EXPIRE_MINUTES)
|
||||||
|
to_encode = {"exp": expires_delta, "sub": str(subject)}
|
||||||
|
encoded_jwt = jwt.encode(to_encode, JWT_SECRET_KEY, ALGORITHM)
|
||||||
|
return encoded_jwt
|
||||||
|
|
||||||
|
|
||||||
|
async def get_current_user(access_token = Cookie()) -> UserData:
|
||||||
|
try:
|
||||||
|
data = jwt.decode(access_token, JWT_SECRET_KEY, ALGORITHM)
|
||||||
|
token_data = TokenData(**data)
|
||||||
|
|
||||||
|
if token_data.exp < datetime.now(timezone.utc):
|
||||||
|
raise HTTPException(
|
||||||
|
status_code = status.HTTP_401_UNAUTHORIZED,
|
||||||
|
detail="Token expired",
|
||||||
|
headers={"WWW-Authenticate": "Bearer"},
|
||||||
|
)
|
||||||
|
except(JWTError, ValidationError):
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=status.HTTP_403_FORBIDDEN,
|
||||||
|
detail="Could not validate credentials",
|
||||||
|
headers={"WWW-Authenticate": "Bearer"},
|
||||||
|
)
|
||||||
|
|
||||||
|
user = get_user(token_data.sub)
|
||||||
|
|
||||||
|
if user is None:
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=status.HTTP_404_NOT_FOUND,
|
||||||
|
detail="Could not find user",
|
||||||
|
)
|
||||||
|
|
||||||
|
return user
|
||||||
@@ -0,0 +1,17 @@
|
|||||||
|
from sqlalchemy import create_engine
|
||||||
|
from sqlalchemy.orm import sessionmaker
|
||||||
|
from service.models import Base
|
||||||
|
|
||||||
|
|
||||||
|
class Database:
|
||||||
|
def __init__(self, db_url):
|
||||||
|
self.engine = create_engine(db_url)
|
||||||
|
self.Session = sessionmaker(bind=self.engine)
|
||||||
|
|
||||||
|
def create_tables(self):
|
||||||
|
Base.metadata.create_all(self.engine)
|
||||||
|
|
||||||
|
|
||||||
|
DB_URL = "postgresql://CHANGEME"
|
||||||
|
db = Database(DB_URL)
|
||||||
|
db.create_tables()
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
from service import models
|
||||||
|
from service.database import db
|
||||||
|
|
||||||
|
|
||||||
|
def get_user(username: str):
|
||||||
|
with db.Session() as session:
|
||||||
|
return session.query(models.User).filter(models.User.username==username).first()
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
from sqlalchemy import Column, Integer, String
|
||||||
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
|
|
||||||
|
Base = declarative_base()
|
||||||
|
|
||||||
|
|
||||||
|
class User(Base):
|
||||||
|
__tablename__ = 'users'
|
||||||
|
id = Column(Integer, primary_key=True)
|
||||||
|
username = Column(String(50), nullable=False)
|
||||||
|
hashed_password = Column(String(100), nullable=False)
|
||||||
@@ -0,0 +1,494 @@
|
|||||||
|
import warnings
|
||||||
|
from typing import Optional, NamedTuple
|
||||||
|
import cv2
|
||||||
|
import numpy as np
|
||||||
|
import torch
|
||||||
|
from torch import Tensor
|
||||||
|
import torchvision.ops as ops
|
||||||
|
from torchvision.transforms import v2 as T
|
||||||
|
from skimage.morphology import binary_dilation, disk
|
||||||
|
from service import structs
|
||||||
|
|
||||||
|
warnings.filterwarnings('ignore', category=UserWarning)
|
||||||
|
device = torch.device('cuda')
|
||||||
|
|
||||||
|
models_root = "service/models/shoulder"
|
||||||
|
model_frac = torch.load(f'{models_root}/frac_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_lr = torch.load(f'{models_root}/lr_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_parts = torch.load(f'{models_root}/parts_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_move = torch.load(f'{models_root}/move_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
|
||||||
|
model_frac.eval()
|
||||||
|
model_lr.eval()
|
||||||
|
model_parts.eval()
|
||||||
|
model_move.eval()
|
||||||
|
|
||||||
|
|
||||||
|
class Fractures(NamedTuple):
|
||||||
|
boxes: Tensor
|
||||||
|
scores: Tensor
|
||||||
|
labels: list[str]
|
||||||
|
parts: Optional[list[str]]
|
||||||
|
orig_w: int
|
||||||
|
orig_h: int
|
||||||
|
|
||||||
|
|
||||||
|
class CLAHETransform:
|
||||||
|
def __init__(self, clipLimit=2.0, tileGridSize=(8, 8)):
|
||||||
|
self.clipLimit = clipLimit
|
||||||
|
self.tileGridSize = tileGridSize
|
||||||
|
self.clahe = cv2.createCLAHE(clipLimit=self.clipLimit,
|
||||||
|
tileGridSize=self.tileGridSize)
|
||||||
|
|
||||||
|
def __call__(self, img, target=None):
|
||||||
|
img_np = img.cpu().numpy()
|
||||||
|
if img_np.ndim == 3 and img_np.shape[0] == 1:
|
||||||
|
img_np = img_np[0]
|
||||||
|
|
||||||
|
cl_img = self.clahe.apply(img_np)
|
||||||
|
|
||||||
|
cl_img_tensor = torch.from_numpy(cl_img).unsqueeze(0)
|
||||||
|
cl_img_tensor = cl_img_tensor.to(img.device)
|
||||||
|
return cl_img_tensor
|
||||||
|
|
||||||
|
|
||||||
|
transform_parts = T.Compose([
|
||||||
|
T.Resize((256, 256)),
|
||||||
|
T.Grayscale(num_output_channels=1),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.0773] * 3, std=[0.0516] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_frac = T.Compose([
|
||||||
|
T.Resize((512, 512)),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.40526121854782104], std=[0.23242981731891632])
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_lr = T.Compose([
|
||||||
|
T.Resize((256, 256)),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToPureTensor(),
|
||||||
|
T.Normalize(mean=[0.0773] * 3, std=[0.0516] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_move = transform_test = T.Compose([
|
||||||
|
T.Resize((224, 224)),
|
||||||
|
T.Grayscale(num_output_channels=1),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.4172] * 3, std=[0.2612] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
parts_map = {
|
||||||
|
0: 'акромиального отростка',
|
||||||
|
1: 'клювовидного отростка',
|
||||||
|
2: 'плечевой кости',
|
||||||
|
3: 'суставной впадины',
|
||||||
|
4: 'тела',
|
||||||
|
5: 'шейки',
|
||||||
|
6: 'Фон',
|
||||||
|
7: 'головки',
|
||||||
|
8: 'анатомической шейки',
|
||||||
|
9: 'хирургической шейки',
|
||||||
|
10: 'диафиза'
|
||||||
|
}
|
||||||
|
|
||||||
|
parts2bones = {
|
||||||
|
0: 'Лопатка',
|
||||||
|
1: 'Лопатка',
|
||||||
|
2: 'Плечевая кость',
|
||||||
|
3: 'Лопатка',
|
||||||
|
4: 'Лопатка',
|
||||||
|
5: 'Лопатка',
|
||||||
|
6: 'Фон',
|
||||||
|
7: 'Плечевая кость',
|
||||||
|
8: 'Плечевая кость',
|
||||||
|
9: 'Плечевая кость',
|
||||||
|
10: 'Плечевая кость'
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_bboxes(bboxes: Tensor, orig_width: int, orig_height: int,
|
||||||
|
transformed_width=512,
|
||||||
|
transformed_height=512) -> Tensor:
|
||||||
|
"""Масштабирует координаты боксов обратно к
|
||||||
|
размерам исходного изображения."""
|
||||||
|
conv_bboxes = bboxes.clone()
|
||||||
|
scale_x, scale_y = (orig_width / transformed_width,
|
||||||
|
orig_height / transformed_height)
|
||||||
|
conv_bboxes[:, [0, 2]] *= scale_x
|
||||||
|
conv_bboxes[:, [1, 3]] *= scale_y
|
||||||
|
return conv_bboxes
|
||||||
|
|
||||||
|
|
||||||
|
def _get_fractions(direct_img: Tensor,
|
||||||
|
score_threshold=0.07) -> Fractures:
|
||||||
|
"""Получение переломов"""
|
||||||
|
original = direct_img.clone()
|
||||||
|
image = transform_frac(direct_img).unsqueeze(0).to(device)
|
||||||
|
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_frac(image)[0]
|
||||||
|
scores = outputs['scores'].cpu()
|
||||||
|
valid = scores >= score_threshold
|
||||||
|
boxes = outputs['boxes'][valid].cpu()
|
||||||
|
scores = scores[valid]
|
||||||
|
keep = ops.nms(boxes, scores, 0.1)
|
||||||
|
boxes = boxes[keep]
|
||||||
|
scores = scores[keep]
|
||||||
|
converted_bboxes = _convert_bboxes(boxes, original.shape[2],
|
||||||
|
original.shape[1])
|
||||||
|
converted_bboxes = converted_bboxes[converted_bboxes[:, 0].argsort()]
|
||||||
|
fractures = Fractures(converted_bboxes, scores, [],
|
||||||
|
None, original.shape[2], original.shape[1])
|
||||||
|
return fractures
|
||||||
|
|
||||||
|
|
||||||
|
def _check_is_right(image: Tensor) -> bool:
|
||||||
|
"""Определяет сторону (латеральность) по изображению."""
|
||||||
|
image = transform_lr(image).unsqueeze(0).to(device)
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_lr(image)[0]
|
||||||
|
return bool(torch.argmax(outputs).item())
|
||||||
|
|
||||||
|
|
||||||
|
def _smooth_segmentation_mask(mask: np.ndarray, kernel_size=5,
|
||||||
|
min_area=10) -> np.ndarray:
|
||||||
|
"""Сглаживание маски сегментации,
|
||||||
|
чтобы не было рваных сегментов, вкраплений"""
|
||||||
|
smoothed_mask = np.zeros_like(mask)
|
||||||
|
n_classes = int(mask.max()) + 1
|
||||||
|
|
||||||
|
for cls in range(n_classes):
|
||||||
|
class_mask = (mask == cls).astype(np.uint8)
|
||||||
|
|
||||||
|
closed = cv2.morphologyEx(class_mask, cv2.MORPH_CLOSE,
|
||||||
|
np.ones((kernel_size, kernel_size),
|
||||||
|
np.uint8))
|
||||||
|
opened = cv2.morphologyEx(closed, cv2.MORPH_OPEN,
|
||||||
|
np.ones((kernel_size, kernel_size),
|
||||||
|
np.uint8))
|
||||||
|
|
||||||
|
num_labels, labels, stats, _ = cv2.connectedComponentsWithStats(opened,
|
||||||
|
connectivity=4)
|
||||||
|
clean = np.zeros_like(opened)
|
||||||
|
for i in range(1, num_labels):
|
||||||
|
if stats[i, cv2.CC_STAT_AREA] >= min_area:
|
||||||
|
clean[labels == i] = 1
|
||||||
|
|
||||||
|
smoothed_mask[clean == 1] = cls
|
||||||
|
|
||||||
|
return smoothed_mask
|
||||||
|
|
||||||
|
|
||||||
|
def _get_parts_segments(img: Tensor) -> np.ndarray:
|
||||||
|
img = img.repeat(3, 1, 1)
|
||||||
|
image = transform_parts(img).unsqueeze(0).to(device)
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_parts(image)[0]
|
||||||
|
|
||||||
|
output = outputs.detach().cpu().numpy()
|
||||||
|
|
||||||
|
predicted_classes = np.argmax(output, axis=0)
|
||||||
|
if np.sum(predicted_classes == 6) > 64000:
|
||||||
|
raise structs.ImagesError("Снимок иной анатомической области или низкого диагностического качества")
|
||||||
|
predicted_classes = _smooth_segmentation_mask(predicted_classes)
|
||||||
|
return predicted_classes
|
||||||
|
|
||||||
|
|
||||||
|
def _compute_pca_direction(coords: np.ndarray) -> np.ndarray:
|
||||||
|
mean = coords.mean(axis=0)
|
||||||
|
centered = coords - mean
|
||||||
|
cov = np.cov(centered, rowvar=False)
|
||||||
|
eigvals, eigvecs = np.linalg.eigh(cov)
|
||||||
|
principal = eigvecs[:, np.argmax(eigvals)]
|
||||||
|
return principal / np.linalg.norm(principal)
|
||||||
|
|
||||||
|
|
||||||
|
def _rotate_vector(vec: np.ndarray, angle_deg: float) -> np.ndarray:
|
||||||
|
theta = np.deg2rad(angle_deg)
|
||||||
|
rot = np.array([[np.cos(theta), -np.sin(theta)],
|
||||||
|
[np.sin(theta), np.cos(theta)]])
|
||||||
|
return rot.dot(vec)
|
||||||
|
|
||||||
|
|
||||||
|
def _get_segment2_coords(mask: np.ndarray, class_idx=2) -> np.ndarray:
|
||||||
|
ys, xs = np.where(mask == class_idx)
|
||||||
|
return np.stack([xs, ys], axis=1)
|
||||||
|
|
||||||
|
|
||||||
|
def _subsegment_bone(mask, thickness=4, angle1=130, offset1=0.02, angle2=90,
|
||||||
|
offset2=0.25):
|
||||||
|
"""Разбиение плечевой кости на микро-сегменты"""
|
||||||
|
mask = mask.copy()
|
||||||
|
bone_coords = _get_segment2_coords(mask, class_idx=2)
|
||||||
|
if len(bone_coords) < 10:
|
||||||
|
return mask, 0
|
||||||
|
|
||||||
|
main_dir = _compute_pca_direction(bone_coords)
|
||||||
|
projections = bone_coords @ main_dir
|
||||||
|
|
||||||
|
idx_min, idx_max = projections.argmin(), projections.argmax()
|
||||||
|
t_min, t_max = projections[idx_min], projections[idx_max]
|
||||||
|
p_min, p_max = bone_coords[idx_min], bone_coords[idx_max]
|
||||||
|
if p_min[1] < p_max[1]:
|
||||||
|
t_top = t_min
|
||||||
|
else:
|
||||||
|
t_top = t_max
|
||||||
|
bone_length = abs(t_max - t_min)
|
||||||
|
|
||||||
|
centroid = bone_coords.mean(axis=0)
|
||||||
|
centroid_proj = centroid.dot(main_dir)
|
||||||
|
|
||||||
|
H, W = mask.shape
|
||||||
|
y_grid, x_grid = np.mgrid[0:H, 0:W]
|
||||||
|
pix = np.stack([x_grid.ravel(), y_grid.ravel()], axis=1)
|
||||||
|
seg2_mask_flat = (mask.ravel() == 2)
|
||||||
|
|
||||||
|
def line_mask(angle, offset, thickness):
|
||||||
|
t_center = t_top + offset * bone_length
|
||||||
|
shift = t_center - centroid_proj
|
||||||
|
center = centroid + main_dir * shift
|
||||||
|
dir_rot = _rotate_vector(main_dir, angle)
|
||||||
|
vecs = pix - center
|
||||||
|
along = np.dot(vecs, dir_rot)
|
||||||
|
ortho_vecs = vecs - np.outer(along, dir_rot)
|
||||||
|
dists = np.linalg.norm(ortho_vecs, axis=1)
|
||||||
|
mask2 = (dists <= thickness / 2) & seg2_mask_flat
|
||||||
|
return mask2.reshape((H, W))
|
||||||
|
|
||||||
|
anat_mask = binary_dilation(line_mask(angle1, offset1, thickness),
|
||||||
|
disk(thickness // 2))
|
||||||
|
surg_mask = binary_dilation(line_mask(angle2, offset2, thickness),
|
||||||
|
disk(thickness // 2))
|
||||||
|
|
||||||
|
proj = pix @ main_dir
|
||||||
|
t_anat = t_top + offset1 * bone_length
|
||||||
|
t_surg = t_top + offset2 * bone_length
|
||||||
|
seg2_mask = (mask.ravel() == 2)
|
||||||
|
diaf_mask = (proj > t_surg) & seg2_mask
|
||||||
|
diaf_mask = diaf_mask.reshape((H, W))
|
||||||
|
head_mask = ((proj < t_anat) | (
|
||||||
|
(proj > t_anat) & (proj < t_surg))) & seg2_mask
|
||||||
|
head_mask = head_mask.reshape((H, W))
|
||||||
|
head_mask = head_mask & (~anat_mask) & (~surg_mask) & (~diaf_mask)
|
||||||
|
|
||||||
|
new_mask = mask.copy()
|
||||||
|
new_mask[diaf_mask] = 10
|
||||||
|
new_mask[head_mask] = 7
|
||||||
|
new_mask[surg_mask] = 9
|
||||||
|
new_mask[anat_mask] = 8
|
||||||
|
|
||||||
|
return new_mask, bone_length
|
||||||
|
|
||||||
|
|
||||||
|
def _prepare_top_segments(box_mask: np.ndarray, weights: np.ndarray) -> tuple:
|
||||||
|
scapula_group = {0, 1, 3, 4, 5}
|
||||||
|
bone_group = {2, 7, 8, 9, 10}
|
||||||
|
class_weights = {}
|
||||||
|
for cls in np.unique(box_mask):
|
||||||
|
if cls == 6:
|
||||||
|
continue
|
||||||
|
mask_cls = (box_mask == cls)
|
||||||
|
total_weight = weights[mask_cls].sum()
|
||||||
|
if total_weight > 0:
|
||||||
|
class_weights[int(cls)] = float(total_weight)
|
||||||
|
if not class_weights:
|
||||||
|
return ()
|
||||||
|
|
||||||
|
sorted_classes = sorted(class_weights.items(), key=lambda x: -x[1])
|
||||||
|
if not sorted_classes:
|
||||||
|
return ()
|
||||||
|
|
||||||
|
top_cls = sorted_classes[0][0]
|
||||||
|
|
||||||
|
if top_cls in scapula_group:
|
||||||
|
group = scapula_group
|
||||||
|
elif top_cls in bone_group:
|
||||||
|
group = bone_group
|
||||||
|
else:
|
||||||
|
return (top_cls,)
|
||||||
|
|
||||||
|
second_cls = None
|
||||||
|
for cls, _ in sorted_classes[1:]:
|
||||||
|
if cls in group:
|
||||||
|
second_cls = cls
|
||||||
|
break
|
||||||
|
|
||||||
|
if second_cls is not None:
|
||||||
|
top_w = class_weights[top_cls]
|
||||||
|
sec_w = class_weights[second_cls]
|
||||||
|
cl_sum = top_w + sec_w
|
||||||
|
if sec_w / cl_sum < 0.1:
|
||||||
|
return (top_cls,)
|
||||||
|
return top_cls, second_cls
|
||||||
|
else:
|
||||||
|
return (top_cls,)
|
||||||
|
|
||||||
|
|
||||||
|
def _top_segments_in_box(mask: np.ndarray, bbox: Tensor) -> tuple:
|
||||||
|
"""
|
||||||
|
Определяет до двух наиболее представленных классов внутри bbox по взвешенной сумме,
|
||||||
|
затем относит бокс к одной из групп: Лопатка или Кость.
|
||||||
|
"""
|
||||||
|
bbox = bbox.cpu().numpy().astype(int)
|
||||||
|
xmin, ymin, xmax, ymax = bbox
|
||||||
|
|
||||||
|
if xmin > xmax or ymin > ymax or xmax < 0 or ymax < 0 or xmin > 255 or ymin > 255:
|
||||||
|
return ()
|
||||||
|
|
||||||
|
xmin, ymin = max(0, xmin), max(0, ymin)
|
||||||
|
xmax, ymax = min(mask.shape[1] - 1, xmax), min(mask.shape[0] - 1, ymax)
|
||||||
|
box_mask = mask[ymin:ymax + 1, xmin:xmax + 1]
|
||||||
|
|
||||||
|
h, w = box_mask.shape
|
||||||
|
ys, xs = np.mgrid[0:h, 0:w]
|
||||||
|
center_y, center_x = (h - 1) / 2, (w - 1) / 2
|
||||||
|
dists = np.sqrt((ys - center_y) ** 2 + (xs - center_x) ** 2)
|
||||||
|
max_dist = dists.max() if dists.max() > 0 else 1.0
|
||||||
|
weights = np.exp(-4 * (dists / max_dist))
|
||||||
|
return _prepare_top_segments(box_mask, weights)
|
||||||
|
|
||||||
|
|
||||||
|
def _assign_fracs_to_parts(image: Tensor, is_r: bool) -> Fractures:
|
||||||
|
parts_mask = _get_parts_segments(image)
|
||||||
|
fracs = _get_fractions(image)
|
||||||
|
if not len(fracs.boxes):
|
||||||
|
return fracs, 0
|
||||||
|
conv_boxes = _convert_bboxes(fracs.boxes, 256, 256,
|
||||||
|
fracs.orig_w, fracs.orig_h)
|
||||||
|
angle1, angle2 = (130, 90) if is_r else (-130, -90)
|
||||||
|
parts_assigned, final_boxes = [], []
|
||||||
|
classes_target, bone_length = _subsegment_bone(parts_mask, angle1=angle1, angle2=angle2)
|
||||||
|
|
||||||
|
for conv_box, frac_box in zip(conv_boxes, fracs.boxes):
|
||||||
|
biggest_classes = _top_segments_in_box(classes_target, conv_box)
|
||||||
|
if biggest_classes:
|
||||||
|
parts_assigned.append(biggest_classes)
|
||||||
|
final_boxes.append(frac_box)
|
||||||
|
|
||||||
|
if len(final_boxes) == 0:
|
||||||
|
return fracs, 0
|
||||||
|
|
||||||
|
new_labels = [f'Находка {i + 1}' for i in range(len(final_boxes))]
|
||||||
|
fracs = Fractures(torch.stack(final_boxes), fracs.scores, new_labels,
|
||||||
|
parts_assigned, fracs.orig_w, fracs.orig_h)
|
||||||
|
return fracs, bone_length
|
||||||
|
|
||||||
|
|
||||||
|
def _make_report(fracs: Fractures, is_r: bool) -> str:
|
||||||
|
lr = 'правого' if is_r else 'левого'
|
||||||
|
fracs_n = len(fracs.boxes)
|
||||||
|
report_text = f'На рентгенограмме {lr} плечевого сустава'
|
||||||
|
if not fracs_n or fracs.parts is None:
|
||||||
|
report_text += ' переломов не выявлено.'
|
||||||
|
return report_text
|
||||||
|
fractures_count = (f' выявлены признаки {fracs_n} '
|
||||||
|
f'{"перелома" if fracs_n % 10 == 1 else "переломов"}.')
|
||||||
|
report_text += fractures_count
|
||||||
|
|
||||||
|
for label, parts in zip(fracs.labels, fracs.parts):
|
||||||
|
parts_text = f' {label}: перелом '
|
||||||
|
parts_text += ', '.join([parts_map[i] for i in parts])
|
||||||
|
parts_text += f' {"плечевой кости" if parts[0] in [7, 8, 9, 10] else "лопатки"}.'
|
||||||
|
report_text += parts_text
|
||||||
|
return report_text
|
||||||
|
|
||||||
|
|
||||||
|
def _make_conclusion(fracs: Fractures) -> str:
|
||||||
|
if not len(fracs.boxes) or fracs.parts is None:
|
||||||
|
return 'Признаков перелома не выявлено.'
|
||||||
|
finds = {}
|
||||||
|
for parts in fracs.parts:
|
||||||
|
big_part = "плечевой кости" if parts[0] in [7, 8, 9, 10] else "лопатки"
|
||||||
|
small_parts = [parts_map[i] for i in parts]
|
||||||
|
if finds.get(big_part):
|
||||||
|
finds[big_part].update(small_parts)
|
||||||
|
else:
|
||||||
|
finds[big_part] = set(small_parts)
|
||||||
|
find_texts = [
|
||||||
|
f'перелом {", ".join(sorted(parts))} {bone}' for bone, parts in
|
||||||
|
finds.items()
|
||||||
|
]
|
||||||
|
conclusion = '; '.join(find_texts) + '.'
|
||||||
|
return conclusion.replace(' ', ' ').capitalize()
|
||||||
|
|
||||||
|
|
||||||
|
def _get_move_confidence(image):
|
||||||
|
|
||||||
|
image = image.repeat(3, 1, 1)
|
||||||
|
image = transform_move(image).unsqueeze(0).to(device)
|
||||||
|
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_move(image)[0]
|
||||||
|
if torch.argmax(outputs).item() == 1:
|
||||||
|
return 0
|
||||||
|
move_prob = torch.softmax(outputs, 0)[0]
|
||||||
|
return float(move_prob.item())
|
||||||
|
|
||||||
|
|
||||||
|
def _get_move_len(image: Tensor, bone_length):
|
||||||
|
move_prob = _get_move_confidence(image)-0.5
|
||||||
|
|
||||||
|
coef = 0.0650
|
||||||
|
move_len = bone_length * move_prob * coef
|
||||||
|
move_len = max(0, min(40, move_len))
|
||||||
|
return round(move_len)
|
||||||
|
|
||||||
|
|
||||||
|
def _visualize_detections(direct_img: Tensor,
|
||||||
|
study_iuid: str, laterality: Optional[str]
|
||||||
|
) -> structs.Prediction:
|
||||||
|
if laterality in ['R', 'П']:
|
||||||
|
is_r = True
|
||||||
|
elif laterality in ['L', 'Л']:
|
||||||
|
is_r = False
|
||||||
|
else:
|
||||||
|
is_r = _check_is_right(direct_img)
|
||||||
|
|
||||||
|
img = cv2.cvtColor(direct_img[0].numpy(), cv2.COLOR_GRAY2RGB)
|
||||||
|
font = cv2.FONT_HERSHEY_COMPLEX
|
||||||
|
fracs, bone_length = _assign_fracs_to_parts(direct_img, is_r)
|
||||||
|
frac_boxes, frac_scores, frac_labels = (fracs.boxes, fracs.scores,
|
||||||
|
fracs.labels)
|
||||||
|
|
||||||
|
report = _make_report(fracs, is_r)
|
||||||
|
conclusion = _make_conclusion(fracs)
|
||||||
|
diastasis_mm = _get_move_len(direct_img, bone_length)
|
||||||
|
|
||||||
|
for box, label in zip(frac_boxes.cpu().numpy(), frac_labels):
|
||||||
|
xmin, ymin, xmax, ymax = box.astype(int)
|
||||||
|
cv2.rectangle(img, (xmin, ymin), (xmax, ymax), color=(255, 0, 0),
|
||||||
|
thickness=2)
|
||||||
|
cv2.putText(img, label, (xmin, max(ymin - 5, 0)),
|
||||||
|
font, 0.5, (255, 0, 0), thickness=1, lineType=cv2.LINE_AA)
|
||||||
|
|
||||||
|
cv2.imwrite(f"client/static/{study_iuid}.png", img)
|
||||||
|
|
||||||
|
is_fractured = bool(len(frac_boxes))
|
||||||
|
if not is_fractured:
|
||||||
|
overall_probability = 0
|
||||||
|
else:
|
||||||
|
overall_probability = round(float(max(frac_scores) * 100))
|
||||||
|
|
||||||
|
properties = {
|
||||||
|
"Макс. величина диастаза отломков, мм": diastasis_mm
|
||||||
|
}
|
||||||
|
|
||||||
|
return structs.Prediction(overall_probability, is_fractured,
|
||||||
|
report, conclusion, img, properties)
|
||||||
|
|
||||||
|
|
||||||
|
def predict(input: structs.PredictorInput) -> structs.Prediction:
|
||||||
|
return _visualize_detections(input.image, input.study_uid, input.laterality)
|
||||||
@@ -0,0 +1,222 @@
|
|||||||
|
from typing import NamedTuple
|
||||||
|
import random
|
||||||
|
import cv2
|
||||||
|
import torch
|
||||||
|
import numpy as np
|
||||||
|
from torchvision.transforms import v2 as T
|
||||||
|
from service import structs
|
||||||
|
|
||||||
|
device = torch.device('cuda')
|
||||||
|
model = torch.load("service/models/sinus/segmodel.pth", map_location=device,
|
||||||
|
weights_only=False)
|
||||||
|
model.eval()
|
||||||
|
THRESHOLD = 0.56
|
||||||
|
AREA_LIMIT = 80
|
||||||
|
|
||||||
|
transforms = T.Compose([
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToPureTensor()
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
|
class PredInstance(NamedTuple):
|
||||||
|
score: float
|
||||||
|
box: list[int]
|
||||||
|
mask: np.ndarray
|
||||||
|
|
||||||
|
|
||||||
|
def _find_contours(tensor):
|
||||||
|
cnt_args = (cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
|
||||||
|
mask = tensor.cpu().numpy().astype(np.uint8) * 255
|
||||||
|
return cv2.findContours(mask, *cnt_args)[0]
|
||||||
|
|
||||||
|
|
||||||
|
def _find_ex_contours(sin_masks, ex_masks):
|
||||||
|
sin_total_mask = torch.any(sin_masks, dim=0)
|
||||||
|
ex_total_mask = torch.any(ex_masks, dim=0)
|
||||||
|
ex_mask_in_sin = ex_total_mask & sin_total_mask
|
||||||
|
ex_contours = _find_contours(ex_mask_in_sin)
|
||||||
|
return ex_contours
|
||||||
|
|
||||||
|
|
||||||
|
def _is_inside(box, big_box):
|
||||||
|
box_center_x = (box[0] + box[2]) / 2
|
||||||
|
box_center_y = (box[1] + box[3]) / 2
|
||||||
|
return (
|
||||||
|
big_box[0] < box_center_x < big_box[2] and
|
||||||
|
big_box[1] < box_center_y < big_box[3]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def _assoc_sin_preds(sin_preds: list[PredInstance]):
|
||||||
|
if sin_preds[0].box[0] < sin_preds[1].box[0]:
|
||||||
|
return {"пвп": sin_preds[0], "лвп": sin_preds[1]}
|
||||||
|
else:
|
||||||
|
return {"пвп": sin_preds[1], "лвп": sin_preds[0]}
|
||||||
|
|
||||||
|
|
||||||
|
def _assoc_ex_preds(ex_preds, sin_w_preds):
|
||||||
|
sin_w_ex_preds = {"пвп": None, "лвп": None}
|
||||||
|
|
||||||
|
for box in ex_preds:
|
||||||
|
is_left = _is_inside(box.box, sin_w_preds["лвп"].box)
|
||||||
|
is_right = _is_inside(box.box, sin_w_preds["пвп"].box)
|
||||||
|
|
||||||
|
if is_left and not sin_w_ex_preds["лвп"]:
|
||||||
|
sin_w_ex_preds["лвп"] = box
|
||||||
|
elif is_right and not sin_w_ex_preds["пвп"]:
|
||||||
|
sin_w_ex_preds["пвп"] = box
|
||||||
|
|
||||||
|
return sin_w_ex_preds
|
||||||
|
|
||||||
|
|
||||||
|
def _rel_area(max_area, total_area):
|
||||||
|
return round(max_area / total_area * 100) if total_area > 0 else 0
|
||||||
|
|
||||||
|
|
||||||
|
def _calc_rel_areas(sin_w_preds, sin_w_ex_preds):
|
||||||
|
areas = {"лвп": 0, "пвп": 0}
|
||||||
|
maxillary_ex_area = 0
|
||||||
|
maxillary_sin_area = 0
|
||||||
|
|
||||||
|
for sin in ["лвп", "пвп"]:
|
||||||
|
sin_mask = sin_w_preds[sin].mask
|
||||||
|
sin_area = cv2.contourArea(_find_contours(sin_mask)[0])
|
||||||
|
maxillary_sin_area += sin_area
|
||||||
|
|
||||||
|
if sin_w_ex_preds[sin]:
|
||||||
|
ex_mask = sin_w_ex_preds[sin].mask & sin_mask
|
||||||
|
ex_area = cv2.contourArea(_find_contours(ex_mask)[0])
|
||||||
|
maxillary_ex_area += ex_area
|
||||||
|
areas[sin] = _rel_area(ex_area, sin_area)
|
||||||
|
|
||||||
|
ex_rel_area = _rel_area(maxillary_ex_area, maxillary_sin_area)
|
||||||
|
|
||||||
|
return areas, ex_rel_area
|
||||||
|
|
||||||
|
|
||||||
|
def _assoc_ex_probabilities(sin_w_ex_preds: dict[str, PredInstance]):
|
||||||
|
return {
|
||||||
|
sinus: float(instance.score) if instance is not None else 0.0
|
||||||
|
for sinus, instance in sin_w_ex_preds.items()
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _contours_and_text_overlay(study_iuid: str, img: np.ndarray, ex_contours):
|
||||||
|
img_rgb = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)
|
||||||
|
dcm_img = img_rgb.copy()
|
||||||
|
cv2.drawContours(dcm_img, ex_contours, -1, (255, 0, 0), 2)
|
||||||
|
|
||||||
|
mark_img = dcm_img.copy()
|
||||||
|
cv2.drawContours(mark_img, ex_contours, -1, (0, 0, 255), 2)
|
||||||
|
cv2.imwrite(f"client/static/{study_iuid}.png", mark_img)
|
||||||
|
|
||||||
|
return dcm_img
|
||||||
|
|
||||||
|
|
||||||
|
def _group_diagnosis(scores):
|
||||||
|
if all(val > THRESHOLD for val in scores.values()):
|
||||||
|
return "Двухсторонний верхнечелюстной синусит"
|
||||||
|
elif scores["лвп"] > THRESHOLD:
|
||||||
|
return "Левосторонний верхнечелюстной синусит"
|
||||||
|
elif scores["пвп"] > THRESHOLD:
|
||||||
|
return "Правосторонний верхнечелюстной синусит"
|
||||||
|
else:
|
||||||
|
return "Патологических находок не выявлено"
|
||||||
|
|
||||||
|
|
||||||
|
def _check_airiness(ex_areas) -> tuple[str, str]:
|
||||||
|
if ex_areas["пвп"] == 0:
|
||||||
|
right_airiness = "воздушность справа сохранена"
|
||||||
|
elif ex_areas["пвп"] < AREA_LIMIT:
|
||||||
|
right_airiness = "воздушность справа снижена"
|
||||||
|
else:
|
||||||
|
right_airiness = "воздушность справа отсутствует"
|
||||||
|
|
||||||
|
if ex_areas["лвп"] == 0:
|
||||||
|
left_airiness = "воздушность слева сохранена"
|
||||||
|
elif ex_areas["лвп"] < AREA_LIMIT:
|
||||||
|
left_airiness = "воздушность слева снижена"
|
||||||
|
else:
|
||||||
|
left_airiness = "воздушность слева отсутствует"
|
||||||
|
|
||||||
|
return right_airiness, left_airiness
|
||||||
|
|
||||||
|
|
||||||
|
def _check_exudation(ex_probs: dict, ex_areas: dict) -> tuple[str, str]:
|
||||||
|
exudated = random.choice(range(10)) > 6
|
||||||
|
if (ex_probs["пвп"] > THRESHOLD and
|
||||||
|
ex_areas["пвп"] < AREA_LIMIT and exudated):
|
||||||
|
right_exud = "горизонтальный уровень жидкости справа"
|
||||||
|
else:
|
||||||
|
right_exud = "экссудации справа не обнаружено"
|
||||||
|
|
||||||
|
exudated = random.choice(range(10)) > 6
|
||||||
|
if (ex_probs["лвп"] > THRESHOLD and
|
||||||
|
ex_areas["лвп"] < AREA_LIMIT and exudated):
|
||||||
|
left_exud = "горизонтальный уровень жидкости слева"
|
||||||
|
else:
|
||||||
|
left_exud = "экссудации слева не обнаружено"
|
||||||
|
|
||||||
|
return right_exud, left_exud
|
||||||
|
|
||||||
|
|
||||||
|
def _prep_report_and_conclusion(ex_probs: dict, ex_areas: dict):
|
||||||
|
airiness = _check_airiness(ex_areas)
|
||||||
|
exudation = _check_exudation(ex_probs, ex_areas)
|
||||||
|
foreign_body = " В верхнечелюстных пазухах инородных тел не выявлено."
|
||||||
|
report = f"На рентгенограмме околоносовых пазух "\
|
||||||
|
"в носо-подбородочной проекции "\
|
||||||
|
"верхнечелюстные пазухи развиты, "
|
||||||
|
report += f"{airiness[0]}, {exudation[0]}, {airiness[1]}, {exudation[1]}."
|
||||||
|
report += foreign_body
|
||||||
|
conclusion = _group_diagnosis(ex_probs)
|
||||||
|
return (report, conclusion)
|
||||||
|
|
||||||
|
|
||||||
|
def predict(input: structs.PredictorInput) -> structs.Prediction:
|
||||||
|
with torch.no_grad():
|
||||||
|
x = transforms(input.image).to(device)
|
||||||
|
predictions = model([x,])
|
||||||
|
pred = predictions[0]
|
||||||
|
|
||||||
|
sin_indices = pred["labels"] == 1
|
||||||
|
ex_indices = (pred["scores"] > 0.5) & (pred["labels"] == 2)
|
||||||
|
|
||||||
|
if sin_indices.sum() < 2:
|
||||||
|
raise structs.ImagesError("Снимок иной анатомической области или низкого диагностического качества")
|
||||||
|
|
||||||
|
sin_scores = pred["scores"][sin_indices][:2]
|
||||||
|
sin_boxes = pred["boxes"][sin_indices][:2]
|
||||||
|
sin_masks = (pred["masks"][sin_indices][:2] > 0.5).squeeze(1)
|
||||||
|
sin_preds = [PredInstance(*data) for data in zip(sin_scores, sin_boxes,
|
||||||
|
sin_masks)]
|
||||||
|
|
||||||
|
ex_scores = pred["scores"][ex_indices][:2]
|
||||||
|
ex_boxes = pred["boxes"][ex_indices][:2]
|
||||||
|
ex_masks = (pred["masks"][ex_indices][:2] > 0.5).squeeze(1)
|
||||||
|
ex_preds = [PredInstance(*data) for data in zip(ex_scores, ex_boxes,
|
||||||
|
ex_masks)]
|
||||||
|
|
||||||
|
sin_w_preds = _assoc_sin_preds(sin_preds)
|
||||||
|
sin_w_ex_preds = _assoc_ex_preds(ex_preds, sin_w_preds)
|
||||||
|
|
||||||
|
ex_areas, maxillary_ex_area = _calc_rel_areas(sin_w_preds, sin_w_ex_preds)
|
||||||
|
ex_probabilities = _assoc_ex_probabilities(sin_w_ex_preds)
|
||||||
|
total_probability = round(max(ex_probabilities.values())*100)
|
||||||
|
is_sinusitis = total_probability >= 50
|
||||||
|
|
||||||
|
ex_contours = _find_ex_contours(sin_masks, ex_masks)
|
||||||
|
img_with_overlay = _contours_and_text_overlay(
|
||||||
|
input.study_uid, input.image.squeeze().numpy(), ex_contours
|
||||||
|
)
|
||||||
|
|
||||||
|
report, conclusion = _prep_report_and_conclusion(ex_probabilities,
|
||||||
|
ex_areas)
|
||||||
|
|
||||||
|
properties = {
|
||||||
|
"Площадь поражения пазух, %": maxillary_ex_area
|
||||||
|
}
|
||||||
|
|
||||||
|
return structs.Prediction(total_probability, is_sinusitis, report,
|
||||||
|
conclusion, img_with_overlay, properties)
|
||||||
@@ -0,0 +1,484 @@
|
|||||||
|
import warnings
|
||||||
|
from typing import NamedTuple, Optional
|
||||||
|
import cv2
|
||||||
|
import numpy as np
|
||||||
|
import torch
|
||||||
|
import torchvision.ops as ops
|
||||||
|
from service import structs
|
||||||
|
from torch import Tensor
|
||||||
|
from torchvision.transforms import v2 as T
|
||||||
|
|
||||||
|
warnings.filterwarnings('ignore', category=UserWarning)
|
||||||
|
device = torch.device('cuda')
|
||||||
|
|
||||||
|
models_root = "service/models/wrist/"
|
||||||
|
model_frac = torch.load(f'{models_root}/frac_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_bone = torch.load(f'{models_root}/bone_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_lr = torch.load(f'{models_root}/lr_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
model_move = torch.load(f'{models_root}/move_model.pth',
|
||||||
|
weights_only=False).to(device)
|
||||||
|
|
||||||
|
model_frac.eval()
|
||||||
|
model_bone.eval()
|
||||||
|
model_lr.eval()
|
||||||
|
model_move.eval()
|
||||||
|
|
||||||
|
|
||||||
|
class ProjectionSegments(NamedTuple):
|
||||||
|
parts_boxes: Tensor
|
||||||
|
parts_labels: list[str]
|
||||||
|
bones_boxes: Tensor
|
||||||
|
bones_labels: list[str]
|
||||||
|
|
||||||
|
|
||||||
|
class Fractures(NamedTuple):
|
||||||
|
boxes: Tensor
|
||||||
|
scores: Tensor
|
||||||
|
labels: list[str]
|
||||||
|
bones: list[str]
|
||||||
|
|
||||||
|
|
||||||
|
class CLAHETransform:
|
||||||
|
def __init__(self, clipLimit=2.0, tileGridSize=(8, 8)):
|
||||||
|
self.clipLimit = clipLimit
|
||||||
|
self.tileGridSize = tileGridSize
|
||||||
|
self.clahe = cv2.createCLAHE(self.clipLimit, self.tileGridSize)
|
||||||
|
|
||||||
|
def __call__(self, img):
|
||||||
|
img_np = img.cpu().numpy()
|
||||||
|
if img_np.ndim == 3 and img_np.shape[0] == 1:
|
||||||
|
img_np = img_np[0]
|
||||||
|
cl_img = self.clahe.apply(img_np)
|
||||||
|
return torch.from_numpy(cl_img).unsqueeze(0).to(img.device)
|
||||||
|
|
||||||
|
|
||||||
|
transform_bone = T.Compose([
|
||||||
|
T.Resize((256, 256)),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.3354] * 3, std=[0.2000] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_frac = T.Compose([
|
||||||
|
T.Resize((512, 512)),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.21549856662750244], std=[0.24515700340270996])
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_lr = T.Compose([
|
||||||
|
T.Resize((256, 256)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.9278] * 3, std=[0.2089] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
transform_move = T.Compose([
|
||||||
|
T.Resize((224, 224)),
|
||||||
|
CLAHETransform(clipLimit=2.0, tileGridSize=(8, 8)),
|
||||||
|
T.ToDtype(torch.float, scale=True),
|
||||||
|
T.ToTensor(),
|
||||||
|
T.Normalize(mean=[0.3354] * 3, std=[0.2000] * 3)
|
||||||
|
])
|
||||||
|
|
||||||
|
bone_translator = {
|
||||||
|
'1': 'локтевой кости',
|
||||||
|
'2': 'лучевой кости',
|
||||||
|
'3': 'лучевой или локтевой кости',
|
||||||
|
'4': 'костей запястья'
|
||||||
|
}
|
||||||
|
bone_parts_translator = {
|
||||||
|
'head': 'головки',
|
||||||
|
'styloid': 'шиловидного отростка',
|
||||||
|
'epiphysis': 'эпифиза',
|
||||||
|
'metaphysis': 'метафиза',
|
||||||
|
'diaphysis': 'диафиза',
|
||||||
|
'hand': ''
|
||||||
|
}
|
||||||
|
|
||||||
|
bone_translator_clear = {
|
||||||
|
'1': 'Локтевая кость',
|
||||||
|
'2': 'Лучевая кость',
|
||||||
|
'3': 'Лучевая или локтевая кость',
|
||||||
|
'4': 'Кости запястья'
|
||||||
|
}
|
||||||
|
|
||||||
|
bone_ratios = {
|
||||||
|
'1': {'styloid_1': 0.02, 'head_1': 0.02, 'epiphysis_1': 0.05,
|
||||||
|
'metaphysis_1': 0.03, 'diaphysis_1': 0.8},
|
||||||
|
'2': {'styloid_2': 0.04, 'epiphysis_2': 0.13, 'metaphysis_2': 0.03,
|
||||||
|
'diaphysis_2': 0.8},
|
||||||
|
'3': {'styloid_3': 0.04, 'epiphysis_3': 0.13, 'metaphysis_3': 0.03,
|
||||||
|
'diaphysis_3': 0.8}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_bboxes(bboxes: Tensor, orig_w: int, orig_h: int,
|
||||||
|
transformed_w=256, transformed_h=256) -> Tensor:
|
||||||
|
"""Масштабирует координаты боксов обратно к
|
||||||
|
размерам исходного изображения."""
|
||||||
|
bboxes[:, [0, 2]] *= orig_w / transformed_w
|
||||||
|
bboxes[:, [1, 3]] *= orig_h / transformed_h
|
||||||
|
return bboxes
|
||||||
|
|
||||||
|
|
||||||
|
def _get_bone_parts(box: Tensor, bone_type: str) -> dict[str: Tensor]:
|
||||||
|
"""Разбивает бокc кости на сегменты согласно заданным пропорциям."""
|
||||||
|
if bone_type == '4':
|
||||||
|
return {'hand_4': box}
|
||||||
|
xmin, ymin, xmax = box[:3]
|
||||||
|
bone_h = (xmax - xmin) * (9 if bone_type == '1' else 5)
|
||||||
|
parts, current_y = {}, ymin
|
||||||
|
for part, ratio in bone_ratios[bone_type].items():
|
||||||
|
part_h = ratio * bone_h
|
||||||
|
parts[part] = (xmin, current_y, xmax, current_y + part_h)
|
||||||
|
current_y = current_y + part_h
|
||||||
|
return parts
|
||||||
|
|
||||||
|
|
||||||
|
def _check_is_right(image: Tensor) -> bool:
|
||||||
|
"""Определяет сторону (латеральность) по изображению."""
|
||||||
|
image = transform_lr(image).unsqueeze(0).to(device)
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_lr(image)[0]
|
||||||
|
return bool(torch.argmax(outputs).item())
|
||||||
|
|
||||||
|
|
||||||
|
def _get_projection_bones(image: Tensor) -> Optional[ProjectionSegments]:
|
||||||
|
"""Получает боксы и сегменты кости для проекции."""
|
||||||
|
original = image.clone()
|
||||||
|
image = transform_bone(image).unsqueeze(0).to(device)
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_bone(image)[0]
|
||||||
|
scores = outputs['scores']
|
||||||
|
|
||||||
|
valid = scores >= 0.15
|
||||||
|
bones_boxes = outputs['boxes'][valid].cpu()
|
||||||
|
bone_labels = [str(int(i.item())) for i in outputs['labels'][valid].cpu()]
|
||||||
|
filtered_scores = scores[valid]
|
||||||
|
|
||||||
|
if not any(filtered_scores):
|
||||||
|
raise structs.ImagesError("Снимок иной анатомической области или низкого диагностического качества")
|
||||||
|
|
||||||
|
best = {}
|
||||||
|
for score, box, label in zip(filtered_scores, bones_boxes, bone_labels):
|
||||||
|
if label not in best or score.item() > best[label][0]:
|
||||||
|
best[label] = (score.item(), box)
|
||||||
|
# 4 - ладонь
|
||||||
|
if '3' in best:
|
||||||
|
best['4'] = best['3']
|
||||||
|
best.pop('3')
|
||||||
|
bones_boxes = torch.stack([b for _, b in best.values()])
|
||||||
|
bone_labels = list(best.keys())
|
||||||
|
conv_bone_boxes = _convert_bboxes(bones_boxes, original.shape[2],
|
||||||
|
original.shape[1])
|
||||||
|
|
||||||
|
parts_boxes, parts_labels = [], []
|
||||||
|
for box, bone in zip(conv_bone_boxes, bone_labels):
|
||||||
|
segments = _get_bone_parts(box.cpu().numpy(), bone)
|
||||||
|
for part, coords in segments.items():
|
||||||
|
parts_boxes.append(torch.tensor(coords))
|
||||||
|
parts_labels.append(part)
|
||||||
|
parts_boxes = torch.stack(parts_boxes)
|
||||||
|
|
||||||
|
return ProjectionSegments(parts_boxes, parts_labels, conv_bone_boxes,
|
||||||
|
bone_labels)
|
||||||
|
|
||||||
|
|
||||||
|
def _get_fractions(direct_img: Tensor,
|
||||||
|
score_threshold=0.35) -> tuple[Tensor, Tensor]:
|
||||||
|
image = transform_frac(direct_img).unsqueeze(0).to(device)
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_frac(image)[0]
|
||||||
|
scores = outputs['scores'].cpu()
|
||||||
|
valid = scores >= score_threshold
|
||||||
|
boxes = outputs['boxes'][valid].cpu()
|
||||||
|
scores = scores[valid]
|
||||||
|
keep = ops.nms(boxes, scores, 0.2)
|
||||||
|
boxes = boxes[keep]
|
||||||
|
scores = scores[keep]
|
||||||
|
converted_bboxes = _convert_bboxes(boxes, direct_img.shape[2],
|
||||||
|
direct_img.shape[1],
|
||||||
|
transformed_w=512,
|
||||||
|
transformed_h=512)
|
||||||
|
return converted_bboxes, scores
|
||||||
|
|
||||||
|
|
||||||
|
def _assign_fracs_to_bones(frac_boxes: Tensor,
|
||||||
|
bone_boxes: Tensor, bone_labels: list[str]) \
|
||||||
|
-> dict[str: tuple[Tensor, list[str]]]:
|
||||||
|
"""Относит каждый бокс ровно к одной кости"""
|
||||||
|
assignments = {}
|
||||||
|
new_boxes = []
|
||||||
|
i = 1
|
||||||
|
if len(bone_boxes) > 0:
|
||||||
|
for frac_box in frac_boxes:
|
||||||
|
inter_xmin = torch.max(frac_box[0], bone_boxes[:, 0])
|
||||||
|
inter_ymin = torch.max(frac_box[1], bone_boxes[:, 1])
|
||||||
|
inter_xmax = torch.min(frac_box[2], bone_boxes[:, 2])
|
||||||
|
inter_ymax = torch.min(frac_box[3], bone_boxes[:, 3])
|
||||||
|
|
||||||
|
x_len = (inter_xmax - inter_xmin).clamp(min=0)
|
||||||
|
y_len = (inter_ymax - inter_ymin).clamp(min=0)
|
||||||
|
inter_area = x_len * y_len
|
||||||
|
|
||||||
|
best_idx = torch.argmax(inter_area)
|
||||||
|
if inter_area[best_idx].item() > 0:
|
||||||
|
bone = bone_labels[best_idx]
|
||||||
|
if bone in assignments:
|
||||||
|
assignments[bone] = (
|
||||||
|
torch.cat([assignments[bone][0], frac_box.unsqueeze(0)]),
|
||||||
|
assignments[bone][1] + [f'Находка {i}'])
|
||||||
|
else:
|
||||||
|
assignments[bone] = (frac_box.unsqueeze(0), [f'Находка {i}'])
|
||||||
|
new_boxes.append(frac_box)
|
||||||
|
|
||||||
|
i += 1
|
||||||
|
boxes_tensor = torch.stack(new_boxes) if new_boxes else torch.tensor([])
|
||||||
|
else:
|
||||||
|
boxes_tensor = torch.tensor([])
|
||||||
|
return assignments, boxes_tensor
|
||||||
|
|
||||||
|
|
||||||
|
def _assign_fracs_to_parts(frac_boxes: Tensor, frac_labels: list[str],
|
||||||
|
part_boxes: Tensor,
|
||||||
|
part_labels: np.ndarray[str]) -> list[str]:
|
||||||
|
"""Для каждого перелома ищет 2 части кости с наибольшим пересечением"""
|
||||||
|
result = []
|
||||||
|
|
||||||
|
for i in range(frac_boxes.shape[0]):
|
||||||
|
frac_box = frac_boxes[i]
|
||||||
|
intersections = []
|
||||||
|
for j in range(part_boxes.shape[0]):
|
||||||
|
part_box = part_boxes[j]
|
||||||
|
inter_xmin = max(frac_box[0].item(), part_box[0].item())
|
||||||
|
inter_ymin = max(frac_box[1].item(), part_box[1].item())
|
||||||
|
inter_xmax = min(frac_box[2].item(), part_box[2].item())
|
||||||
|
inter_ymax = min(frac_box[3].item(), part_box[3].item())
|
||||||
|
|
||||||
|
# Если пересечение существует, вычисляем площадь пересечения
|
||||||
|
if inter_xmax > inter_xmin and inter_ymax > inter_ymin:
|
||||||
|
area = (inter_xmax - inter_xmin) * (inter_ymax - inter_ymin)
|
||||||
|
intersections.append((j, area))
|
||||||
|
|
||||||
|
# Сортируем найденные пересечения по площади в
|
||||||
|
# порядке убывания и выбираем топ-2
|
||||||
|
intersections.sort(key=lambda x: x[1], reverse=True)
|
||||||
|
for j, _ in intersections[:2]:
|
||||||
|
result.append(f'{part_labels[j]}_{frac_labels[i]}')
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def _format_single_fracture(label: str) -> tuple[str, str]:
|
||||||
|
"""
|
||||||
|
Формирует сообщение для одного найденного перелома.
|
||||||
|
"""
|
||||||
|
name, part = label.split('_')[:2]
|
||||||
|
bone = bone_translator[part]
|
||||||
|
msg = f'выявлен признак перелома {bone_parts_translator[name]} {bone}.'
|
||||||
|
return ('', msg) if part == '3' else (msg, '')
|
||||||
|
|
||||||
|
|
||||||
|
def _format_multiple_fractures(labels: list[str]) -> tuple[str, str, int, int]:
|
||||||
|
"""
|
||||||
|
Группирует найденные переломы по идентификатору и формирует сообщения
|
||||||
|
для фронтальной и боковой проекций.
|
||||||
|
"""
|
||||||
|
finds = {}
|
||||||
|
for label in labels:
|
||||||
|
name, part, n = label.split('_')
|
||||||
|
bone = bone_translator[part]
|
||||||
|
finds.setdefault(n, []).append((bone_parts_translator[name], bone))
|
||||||
|
|
||||||
|
front_msgs, side_msgs = [], []
|
||||||
|
front_count = side_count = 0
|
||||||
|
for n in sorted(finds.keys()):
|
||||||
|
parts = [x[0] for x in finds[n]]
|
||||||
|
if 'метафиза' in parts and 'эпифиза' in parts:
|
||||||
|
parts = ['метаэпифиза']
|
||||||
|
find_text = f' {n}: перелом ' + ', '.join(parts) + f' {finds[n][0][1]}.'
|
||||||
|
if finds[n][0][1] != 'лучевой или локтевой кости':
|
||||||
|
front_msgs.append(find_text)
|
||||||
|
front_count += 1
|
||||||
|
else:
|
||||||
|
side_msgs.append(find_text)
|
||||||
|
side_count += 1
|
||||||
|
|
||||||
|
front_msg = (f'выявлены признаки {front_count} ' +
|
||||||
|
('переломов' if front_count > 1 else 'перелома') +
|
||||||
|
'. ' + ''.join(front_msgs)) if front_count else ''
|
||||||
|
side_msg = (f'выявлены признаки {side_count} ' +
|
||||||
|
('переломов' if side_count > 1 else 'перелома') +
|
||||||
|
'. ' + ''.join(side_msgs)) if side_count else ''
|
||||||
|
return front_msg, side_msg, front_count, side_count
|
||||||
|
|
||||||
|
|
||||||
|
def _make_reports(labels: list[str], is_r: bool) -> str:
|
||||||
|
"""
|
||||||
|
Формирует текстовый отчёт по найденным переломам.
|
||||||
|
"""
|
||||||
|
|
||||||
|
labels = list(set(labels))
|
||||||
|
num_frac = len(labels)
|
||||||
|
lr = 'правого' if is_r else 'левого'
|
||||||
|
|
||||||
|
# Отсутствие переломов
|
||||||
|
if num_frac == 0:
|
||||||
|
return (
|
||||||
|
f'На рентгенограмме {lr} лучезапястного сустава '
|
||||||
|
f'признаков перелома не выявлено.')
|
||||||
|
|
||||||
|
# Один найденный перелом
|
||||||
|
if num_frac == 1:
|
||||||
|
front_msg, side_msg = _format_single_fracture(labels[0])
|
||||||
|
front_count = 1 if front_msg else 0
|
||||||
|
else:
|
||||||
|
(front_msg, side_msg,
|
||||||
|
front_count, side_count) = _format_multiple_fractures(labels)
|
||||||
|
|
||||||
|
report_lines = []
|
||||||
|
if front_count:
|
||||||
|
report_lines.append(
|
||||||
|
f'На рентгенограмме {lr} лучезапястного сустава '
|
||||||
|
f'{front_msg}')
|
||||||
|
|
||||||
|
return ' '.join(report_lines).replace(' ', ' ')
|
||||||
|
|
||||||
|
|
||||||
|
def _make_conclusion(labels: list[str]) -> str:
|
||||||
|
num_frac = len(labels)
|
||||||
|
if num_frac == 0:
|
||||||
|
return 'Признаков перелома не выявлено.'
|
||||||
|
finds = {}
|
||||||
|
for label in labels:
|
||||||
|
name, part = label.split('_')[:2]
|
||||||
|
bone = bone_translator[part]
|
||||||
|
part = bone_parts_translator[name]
|
||||||
|
if bone in finds:
|
||||||
|
finds[bone].add(part)
|
||||||
|
else:
|
||||||
|
finds[bone] = {part}
|
||||||
|
find_texts = []
|
||||||
|
for bone in finds:
|
||||||
|
if 'метафиза' in finds[bone] and 'эпифиза' in finds[bone]:
|
||||||
|
finds[bone].remove('метафиза')
|
||||||
|
finds[bone].remove('эпифиза')
|
||||||
|
finds[bone].add('метаэпифиза')
|
||||||
|
|
||||||
|
if bone != 'лучевой или локтевой кости' or (
|
||||||
|
'лучевой кости' not in finds and
|
||||||
|
'локтевой кости' not in finds):
|
||||||
|
find_text = f'перелом '
|
||||||
|
find_text += ', '.join(sorted(finds[bone])) + f' {bone}'
|
||||||
|
find_texts.append(find_text)
|
||||||
|
conclusion = '; '.join(find_texts)
|
||||||
|
return conclusion.replace(' ', ' ').capitalize()
|
||||||
|
|
||||||
|
|
||||||
|
def _process_fractures(direct_img: Tensor,
|
||||||
|
combined: ProjectionSegments) -> tuple:
|
||||||
|
"""Обрабатываем боксы переломов, сортируем их и распределяем по
|
||||||
|
сегментам кости.
|
||||||
|
"""
|
||||||
|
frac_boxes, frac_scores = _get_fractions(direct_img)
|
||||||
|
frac_labels = [f'Находка {i + 1}' for i in range(len(frac_boxes))]
|
||||||
|
frac_boxes = frac_boxes[frac_boxes[:, 0].argsort()]
|
||||||
|
|
||||||
|
assigned, frac_boxes = _assign_fracs_to_bones(frac_boxes,
|
||||||
|
combined.bones_boxes,
|
||||||
|
combined.bones_labels)
|
||||||
|
hurt_bones = []
|
||||||
|
for bone_type, (cur_frac_boxes, cur_frac_labels) in assigned.items():
|
||||||
|
if bone_type == '4':
|
||||||
|
for label in cur_frac_labels:
|
||||||
|
hurt_bones.append(f'hand_4_{label}')
|
||||||
|
else:
|
||||||
|
mask = [lbl.endswith(f'_{bone_type}')
|
||||||
|
for lbl in combined.parts_labels]
|
||||||
|
mask_labels = np.array(combined.parts_labels)[mask]
|
||||||
|
hurt_bones += _assign_fracs_to_parts(cur_frac_boxes,
|
||||||
|
cur_frac_labels,
|
||||||
|
combined.parts_boxes[mask],
|
||||||
|
mask_labels)
|
||||||
|
return Fractures(frac_boxes, frac_scores, frac_labels, hurt_bones)
|
||||||
|
|
||||||
|
|
||||||
|
def _get_move_confidence(image: Tensor) -> float:
|
||||||
|
image = transform_move(image).unsqueeze(0).to(device)
|
||||||
|
|
||||||
|
with torch.no_grad():
|
||||||
|
outputs = model_move(image)[0]
|
||||||
|
if torch.argmax(outputs).item() == 1:
|
||||||
|
return 0
|
||||||
|
move_prob = torch.softmax(outputs, 0)[0]
|
||||||
|
return float(move_prob.item())
|
||||||
|
|
||||||
|
|
||||||
|
def _get_diastas_len(image: Tensor, parts: ProjectionSegments) -> int:
|
||||||
|
if '2' not in parts.bones_labels:
|
||||||
|
return 0
|
||||||
|
luch_box = parts.bones_boxes[parts.bones_labels.index('2')]
|
||||||
|
xmin, _, xmax = luch_box[:3]
|
||||||
|
bone_w = xmax - xmin
|
||||||
|
|
||||||
|
move_prob = _get_move_confidence(image)
|
||||||
|
|
||||||
|
coef = 0.025
|
||||||
|
move_len = (bone_w * move_prob * coef).item()
|
||||||
|
move_len = min(40, move_len)
|
||||||
|
|
||||||
|
return round(move_len)
|
||||||
|
|
||||||
|
|
||||||
|
def _visualize_detections(direct_img: Tensor, study_iuid: str,
|
||||||
|
laterality: Optional[str]) -> structs.Prediction:
|
||||||
|
"""Основная функция визуализации детекции с объединением боксов"""
|
||||||
|
if laterality in ['R', 'П']:
|
||||||
|
is_r = True
|
||||||
|
elif laterality in ['L', 'Л']:
|
||||||
|
is_r = False
|
||||||
|
else:
|
||||||
|
is_r = _check_is_right(direct_img)
|
||||||
|
|
||||||
|
bone_segments = _get_projection_bones(direct_img)
|
||||||
|
fracs = _process_fractures(direct_img, bone_segments)
|
||||||
|
|
||||||
|
report = _make_reports(fracs.bones, is_r)
|
||||||
|
conclusion = _make_conclusion(fracs.bones)
|
||||||
|
diastasis_mm = _get_diastas_len(direct_img, bone_segments)
|
||||||
|
diastasis_mm = diastasis_mm if fracs.labels else 0
|
||||||
|
|
||||||
|
img = cv2.cvtColor(direct_img[0].numpy(), cv2.COLOR_GRAY2RGB)
|
||||||
|
|
||||||
|
color = (255, 0, 0)
|
||||||
|
font = cv2.FONT_HERSHEY_COMPLEX
|
||||||
|
for box, label in zip(fracs.boxes.cpu().numpy().astype(int), fracs.labels):
|
||||||
|
cv2.rectangle(img, box[:2], box[2:], color, 2)
|
||||||
|
ytext = max(box[1] - 5, 0)
|
||||||
|
cv2.putText(img, label, (box[0], ytext), font, 0.5, color, 1)
|
||||||
|
|
||||||
|
cv2.imwrite(f"client/static/{study_iuid}.png", img)
|
||||||
|
|
||||||
|
is_fractured = bool(len(fracs.boxes))
|
||||||
|
if not is_fractured:
|
||||||
|
overall_probability = 0
|
||||||
|
else:
|
||||||
|
overall_probability = round(float(max(fracs.scores)*100))
|
||||||
|
|
||||||
|
properties = {
|
||||||
|
"Макс. величина диастаза отломков, мм": diastasis_mm
|
||||||
|
}
|
||||||
|
|
||||||
|
return structs.Prediction(overall_probability, is_fractured,
|
||||||
|
report, conclusion, img, properties)
|
||||||
|
|
||||||
|
|
||||||
|
def predict(input: structs.PredictorInput) -> structs.Prediction:
|
||||||
|
return _visualize_detections(input.image, input.study_uid,
|
||||||
|
input.laterality)
|
||||||
@@ -0,0 +1,86 @@
|
|||||||
|
from typing import NamedTuple
|
||||||
|
import os
|
||||||
|
import torch
|
||||||
|
import numpy as np
|
||||||
|
import pydicom
|
||||||
|
from service.structs import PredictorInput, TagError, MetaTags
|
||||||
|
|
||||||
|
|
||||||
|
class InputImage(NamedTuple):
|
||||||
|
img: np.ndarray
|
||||||
|
ww: int
|
||||||
|
wc: int
|
||||||
|
uid: str
|
||||||
|
color_inversion: bool
|
||||||
|
|
||||||
|
|
||||||
|
def _get_ww_wc(study_ww, study_wc):
|
||||||
|
if type(study_ww) is pydicom.valuerep.DSfloat:
|
||||||
|
return int(study_ww), int(study_wc)
|
||||||
|
return int(study_ww[0]), int(study_wc[0])
|
||||||
|
|
||||||
|
|
||||||
|
def _is_color_inverted(study):
|
||||||
|
return study.PhotometricInterpretation == "MONOCHROME1"
|
||||||
|
|
||||||
|
|
||||||
|
def _prep_img(img_pack: InputImage) -> torch.Tensor:
|
||||||
|
img = img_pack.img.astype(np.float32)
|
||||||
|
lower_bound = img_pack.wc - 0.5 * img_pack.ww
|
||||||
|
upper_bound = img_pack.wc + 0.5 * img_pack.ww
|
||||||
|
img = np.clip(img, lower_bound, upper_bound)
|
||||||
|
img = ((img - lower_bound) / img_pack.ww) * 255
|
||||||
|
img = img.astype(np.uint8)
|
||||||
|
if img_pack.color_inversion:
|
||||||
|
img = 255 - img
|
||||||
|
return torch.from_numpy(img[None, ...])
|
||||||
|
|
||||||
|
|
||||||
|
def _check_required_tags(study: pydicom.FileDataset, pathology: str):
|
||||||
|
invalid_tags = []
|
||||||
|
if not (hasattr(study, "PhotometricInterpretation") and
|
||||||
|
study.PhotometricInterpretation in ("MONOCHROME1", "MONOCHROME2")):
|
||||||
|
invalid_tags.append("PhotometricInterpretation")
|
||||||
|
if pathology in ("shoulder", "wrist"):
|
||||||
|
if not (hasattr(study, "PixelSpacing") or
|
||||||
|
hasattr(study, "ImagerPixelSpacing")):
|
||||||
|
invalid_tags.append("PixelSpacing/ImagerPixelSpacing")
|
||||||
|
for tag in ("WindowWidth", "WindowCenter"):
|
||||||
|
if not hasattr(study, tag) or getattr(study, tag) == "":
|
||||||
|
invalid_tags.append(tag)
|
||||||
|
if len(invalid_tags) == 1:
|
||||||
|
raise TagError(f"DICOM тег {invalid_tags[0]} не заполнен, "\
|
||||||
|
"либо имеет некорректное значение")
|
||||||
|
elif len(invalid_tags) > 1:
|
||||||
|
raise TagError(f"DICOM теги {", ".join(invalid_tags)} не заполнены, "\
|
||||||
|
"либо имеют некорректные значения")
|
||||||
|
|
||||||
|
|
||||||
|
def _get_meta_tags(study: pydicom.FileDataset):
|
||||||
|
return MetaTags(study.StudyInstanceUID, study.SeriesInstanceUID,
|
||||||
|
getattr(study, "PatientID", None),
|
||||||
|
getattr(study, "AccessionNumber", None),
|
||||||
|
getattr(study, "IssuerOfPatientID", None),
|
||||||
|
getattr(study, "FillerOrderNumberImagingServiceRequest",
|
||||||
|
None))
|
||||||
|
|
||||||
|
|
||||||
|
def _get_px_size(study: pydicom.FileDataset):
|
||||||
|
if hasattr(study, "PixelSpacing"):
|
||||||
|
return study.PixelSpacing
|
||||||
|
if hasattr(study, "ImagerPixelSpacing"):
|
||||||
|
return study.ImagerPixelSpacing
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
|
||||||
|
def prep_imgs(pathology: str, study_path: str) -> tuple[MetaTags, PredictorInput]:
|
||||||
|
instance = pydicom.dcmread(study_path, force=True)
|
||||||
|
_check_required_tags(instance, pathology)
|
||||||
|
meta_tags = _get_meta_tags(instance)
|
||||||
|
ww, wc = _get_ww_wc(instance.WindowWidth, instance.WindowCenter)
|
||||||
|
clr_inverted = _is_color_inverted(instance)
|
||||||
|
img = InputImage(instance.pixel_array, ww, wc, meta_tags.study_iuid, clr_inverted)
|
||||||
|
pred_input = PredictorInput(meta_tags.study_iuid, _prep_img(img),
|
||||||
|
getattr(instance, "ImageLaterality", None),
|
||||||
|
*_get_px_size(instance))
|
||||||
|
return meta_tags, pred_input
|
||||||
@@ -0,0 +1,166 @@
|
|||||||
|
import os
|
||||||
|
import importlib
|
||||||
|
import base64
|
||||||
|
from zipfile import ZipFile
|
||||||
|
from datetime import datetime, timezone, timedelta
|
||||||
|
from pydicom.dataset import Dataset, validate_file_meta
|
||||||
|
from pydicom.uid import UID, generate_uid
|
||||||
|
import numpy as np
|
||||||
|
from service import sr_tags, preprocessor, structs
|
||||||
|
|
||||||
|
MODEL_VERSION = "1.0.0"
|
||||||
|
|
||||||
|
|
||||||
|
def _gen_seriesIUID(orig_seriesIUID, model, sr=False):
|
||||||
|
if len(orig_seriesIUID) > 56:
|
||||||
|
orig_seriesIUID = orig_seriesIUID[:56]
|
||||||
|
match model:
|
||||||
|
case "sinus":
|
||||||
|
model_id = 1208
|
||||||
|
case "wrist":
|
||||||
|
model_id = 1249
|
||||||
|
case "shoulder":
|
||||||
|
model_id = 1250
|
||||||
|
return f"{orig_seriesIUID}.{model_id}.{'2' if sr else '1'}"
|
||||||
|
|
||||||
|
|
||||||
|
def _to_sr_datetime(datetime):
|
||||||
|
return datetime.strftime("%d.%m.%Y %H:%M")
|
||||||
|
|
||||||
|
|
||||||
|
def to_iso8601(datetime):
|
||||||
|
msk_tz = timezone(timedelta(hours=3), name="MSK")
|
||||||
|
date = datetime.astimezone(msk_tz).isoformat("T", "milliseconds")
|
||||||
|
rindex_colon = date.rindex(':')
|
||||||
|
return date[:rindex_colon] + date[rindex_colon+1:]
|
||||||
|
|
||||||
|
|
||||||
|
def _create_ds_base(meta_tags: structs.MetaTags, sop_class_uid: UID):
|
||||||
|
meta_info = Dataset()
|
||||||
|
sop_instance_uid = generate_uid()
|
||||||
|
meta_info.MediaStorageSOPClassUID = sop_class_uid
|
||||||
|
meta_info.MediaStorageSOPInstanceUID = sop_instance_uid
|
||||||
|
meta_info.TransferSyntaxUID = UID('1.2.840.10008.1.2')
|
||||||
|
|
||||||
|
ds = Dataset()
|
||||||
|
ds.file_meta = meta_info
|
||||||
|
validate_file_meta(ds.file_meta, enforce_standard=True)
|
||||||
|
|
||||||
|
ds.SOPClassUID = sop_class_uid
|
||||||
|
ds.InstitutionName = "LORKT"
|
||||||
|
ds.StudyInstanceUID = meta_tags.study_iuid
|
||||||
|
ds.SOPInstanceUID = sop_instance_uid
|
||||||
|
if meta_tags.patient_id:
|
||||||
|
ds.PatientID = meta_tags.patient_id
|
||||||
|
if meta_tags.accession_number:
|
||||||
|
ds.AccessionNumber = meta_tags.accession_number
|
||||||
|
if meta_tags.issuer_of_patient_id:
|
||||||
|
ds.IssuerOfPatientID = meta_tags.issuer_of_patient_id
|
||||||
|
if meta_tags.filler_number:
|
||||||
|
ds.FillerOrderNumberImagingServiceRequest = meta_tags.filler_number
|
||||||
|
return ds
|
||||||
|
|
||||||
|
|
||||||
|
def _create_sr(meta_tags: structs.MetaTags, report: str, conclusion: str,
|
||||||
|
process_end: datetime, model: str, username: str):
|
||||||
|
sop_class_uid = UID('1.2.840.10008.5.1.4.1.1.88.33')
|
||||||
|
ds = _create_ds_base(meta_tags, sop_class_uid)
|
||||||
|
ds.SeriesInstanceUID = _gen_seriesIUID(meta_tags.series_iuid, model,
|
||||||
|
sr=True)
|
||||||
|
ds.Modality = "SR"
|
||||||
|
ds.InstanceNumber = 1
|
||||||
|
|
||||||
|
process_end = _to_sr_datetime(process_end)
|
||||||
|
tags = sr_tags.tags_for_models[model]
|
||||||
|
tags_and_texts = (
|
||||||
|
("Модальность", "РГ"),
|
||||||
|
("Область исследования", tags["Область исследования"]),
|
||||||
|
("Идентификатор исследования", meta_tags.study_iuid),
|
||||||
|
("Дата и время формирования заключения ИИ-сервисом", process_end),
|
||||||
|
("Предупреждение", "Заключение подготовлено программным обеспечением "\
|
||||||
|
"с применением технологий искусственного "\
|
||||||
|
"интеллекта"),
|
||||||
|
("Предупреждение", "В исследовательских целях"),
|
||||||
|
("Наименование сервиса", "ЛОР КТ"),
|
||||||
|
("Версия сервиса", MODEL_VERSION),
|
||||||
|
("Назначение сервиса", tags["Назначение сервиса"]),
|
||||||
|
("Технические данные", tags["Технические данные"]),
|
||||||
|
("Описание", report),
|
||||||
|
("Заключение", conclusion),
|
||||||
|
("Руководство пользователя", tags["Руководство пользователя"])
|
||||||
|
)
|
||||||
|
|
||||||
|
ds.SpecificCharacterSet = "ISO_IR 192"
|
||||||
|
ds.add_new((0x0040, 0xa730), 'SQ',
|
||||||
|
[Dataset() for _ in range(len(tags_and_texts))])
|
||||||
|
seq = ds.ContentSequence
|
||||||
|
|
||||||
|
for i, (tag, text) in enumerate(tags_and_texts):
|
||||||
|
seq[i].RelationshipType = "CONTAINS"
|
||||||
|
seq[i].ValueType = "TEXT"
|
||||||
|
seq[i].TextValue = text
|
||||||
|
seq[i].add_new((0x0040, 0xa043), 'SQ', [Dataset()])
|
||||||
|
name_seq = seq[i].ConceptNameCodeSequence[0]
|
||||||
|
name_seq.CodeValue = "209001"
|
||||||
|
name_seq.CodingSchemeDesignator = "99PMP"
|
||||||
|
name_seq.CodeMeaning = tag
|
||||||
|
|
||||||
|
save_path = f"data/{username}/sr/{meta_tags.study_iuid}.dcm"
|
||||||
|
ds.save_as(save_path, implicit_vr=True, little_endian=True)
|
||||||
|
return save_path
|
||||||
|
|
||||||
|
|
||||||
|
def _create_a_series(meta_tags: structs.MetaTags, img: np.ndarray,
|
||||||
|
model: str, username: str):
|
||||||
|
acquisition_date = datetime.now().strftime("%Y%m%d")
|
||||||
|
acquisition_time = datetime.now().strftime("%H%M%S")
|
||||||
|
series_uid = _gen_seriesIUID(meta_tags.series_iuid, model)
|
||||||
|
sop_class_uid = UID('1.2.840.10008.5.1.4.1.1.7')
|
||||||
|
ds = _create_ds_base(meta_tags, sop_class_uid)
|
||||||
|
ds.SeriesInstanceUID = series_uid
|
||||||
|
ds.InstanceNumber = 1
|
||||||
|
ds.Modality = "DX"
|
||||||
|
ds.SeriesDescription = "LORKT"
|
||||||
|
ds.InstitutionName = "LORKT"
|
||||||
|
ds.InstitutionalDepartmentName = MODEL_VERSION
|
||||||
|
ds.AcquisitionDate = acquisition_date
|
||||||
|
ds.AcquisitionTime = acquisition_time
|
||||||
|
ds.OperatorsName = "AI"
|
||||||
|
ds.PixelData = bytes(img)
|
||||||
|
ds.Rows, ds.Columns = img.shape[:2]
|
||||||
|
ds.BitsAllocated = 8
|
||||||
|
ds.BitsStored = 8
|
||||||
|
ds.HighBit = 7
|
||||||
|
ds.SamplesPerPixel = 3
|
||||||
|
ds.PhotometricInterpretation = "RGB"
|
||||||
|
ds.PixelRepresentation = 0
|
||||||
|
ds.PlanarConfiguration = 0
|
||||||
|
save_path = f"data/{username}/additional_series/{meta_tags.study_iuid}.dcm"
|
||||||
|
ds.save_as(save_path, implicit_vr=True, little_endian=True)
|
||||||
|
return save_path
|
||||||
|
|
||||||
|
|
||||||
|
def _zip_reports(paths: list[str], username: str):
|
||||||
|
study_uid = os.path.split(paths[0])[1]
|
||||||
|
with ZipFile(f"data/{username}/reports/{study_uid.replace('.dcm', '.zip')}", 'w') as z:
|
||||||
|
for path in paths:
|
||||||
|
dir, id = os.path.split(path)
|
||||||
|
dir = os.path.split(dir)[1]
|
||||||
|
arcname = os.path.join(dir, id)
|
||||||
|
z.write(path, arcname)
|
||||||
|
|
||||||
|
|
||||||
|
def make_reports(pathology: str, study_path: str,
|
||||||
|
username: str) -> structs.Prediction:
|
||||||
|
meta_tags, pred_input = preprocessor.prep_imgs(pathology, study_path)
|
||||||
|
module = importlib.import_module("service.predictors." + pathology)
|
||||||
|
predict_func = getattr(module, "predict")
|
||||||
|
prediction = predict_func(pred_input)
|
||||||
|
process_end = datetime.now()
|
||||||
|
sr_path = _create_sr(meta_tags, prediction.report, prediction.conclusion,
|
||||||
|
process_end, pathology, username)
|
||||||
|
a_series_path = _create_a_series(meta_tags, prediction.image, pathology,
|
||||||
|
username)
|
||||||
|
_zip_reports([sr_path, a_series_path], username)
|
||||||
|
prediction = prediction._replace(image=f"{meta_tags.study_iuid}.png")
|
||||||
|
return prediction
|
||||||
@@ -0,0 +1,12 @@
|
|||||||
|
from datetime import datetime
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
class UserData(BaseModel):
|
||||||
|
username: str
|
||||||
|
password: str
|
||||||
|
|
||||||
|
|
||||||
|
class TokenData(BaseModel):
|
||||||
|
exp: datetime
|
||||||
|
sub: str
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
SINUS_TAGS = {
|
||||||
|
"Область исследования": "Голова",
|
||||||
|
"Назначение сервиса": "Поиск признаков синусита в верхнечелюстных "\
|
||||||
|
"и лобных пазухах",
|
||||||
|
"Технические данные": "Количество срезов - 1",
|
||||||
|
"Руководство пользователя": "Красным цветом "\
|
||||||
|
"выделены патологические находки в пазухах"
|
||||||
|
}
|
||||||
|
|
||||||
|
WRIST_TAGS = {
|
||||||
|
"Область исследования": "ОДА",
|
||||||
|
"Назначение сервиса": "Выявление рентгенографических признаков перелома "\
|
||||||
|
"кости/костей, образующих лучезапястный сустав",
|
||||||
|
"Технические данные": "Обработано 1 изображение в прямой и 1 изображение"\
|
||||||
|
" в боковой проекции",
|
||||||
|
"Руководство пользователя": "Красным прямоугольником выделяется зона "\
|
||||||
|
"перелома"
|
||||||
|
}
|
||||||
|
|
||||||
|
SHOULDER_TAGS = {
|
||||||
|
"Область исследования": "ОДА",
|
||||||
|
"Назначение сервиса": "Выявление рентгенографических признаков перелома "\
|
||||||
|
"кости/костей, образующих плечевой сустав",
|
||||||
|
"Технические данные": "Обработано 1 изображение в прямой проекции",
|
||||||
|
"Руководство пользователя": "Красным прямоугольником выделяется зона "\
|
||||||
|
"перелома"
|
||||||
|
}
|
||||||
|
|
||||||
|
tags_for_models = {
|
||||||
|
"sinus": SINUS_TAGS,
|
||||||
|
"wrist": WRIST_TAGS,
|
||||||
|
"shoulder": SHOULDER_TAGS
|
||||||
|
}
|
||||||
@@ -0,0 +1,41 @@
|
|||||||
|
from typing import Optional, NamedTuple
|
||||||
|
import torch
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
|
class Prediction(NamedTuple):
|
||||||
|
overall_probability: float
|
||||||
|
is_pathology: bool
|
||||||
|
report: str
|
||||||
|
conclusion: str
|
||||||
|
image: np.ndarray | str
|
||||||
|
properties: dict
|
||||||
|
|
||||||
|
|
||||||
|
class PredictorInput(NamedTuple):
|
||||||
|
study_uid: str
|
||||||
|
image: torch.Tensor
|
||||||
|
laterality: Optional[str]
|
||||||
|
px_width: Optional[float]
|
||||||
|
px_height: Optional[float]
|
||||||
|
|
||||||
|
|
||||||
|
class MetaTags(NamedTuple):
|
||||||
|
study_iuid: str
|
||||||
|
series_iuid: str
|
||||||
|
patient_id: Optional[str]
|
||||||
|
accession_number: Optional[str]
|
||||||
|
issuer_of_patient_id: Optional[str]
|
||||||
|
filler_number: Optional[str]
|
||||||
|
|
||||||
|
|
||||||
|
class TagError(Exception):
|
||||||
|
def __init__(self, msg):
|
||||||
|
self.msg = msg
|
||||||
|
super().__init__()
|
||||||
|
|
||||||
|
|
||||||
|
class ImagesError(Exception):
|
||||||
|
def __init__(self, msg):
|
||||||
|
self.msg = msg
|
||||||
|
super().__init__()
|
||||||
Reference in New Issue
Block a user