endoreg-db 0.8.2.9__py3-none-any.whl → 0.8.3.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of endoreg-db might be problematic. Click here for more details.
- endoreg_db/models/media/video/pipe_1.py +13 -4
- endoreg_db/models/metadata/model_meta.py +4 -5
- endoreg_db/services/video_import.py +3 -4
- {endoreg_db-0.8.2.9.dist-info → endoreg_db-0.8.3.1.dist-info}/METADATA +1 -1
- {endoreg_db-0.8.2.9.dist-info → endoreg_db-0.8.3.1.dist-info}/RECORD +7 -7
- {endoreg_db-0.8.2.9.dist-info → endoreg_db-0.8.3.1.dist-info}/WHEEL +0 -0
- {endoreg_db-0.8.2.9.dist-info → endoreg_db-0.8.3.1.dist-info}/licenses/LICENSE +0 -0
|
@@ -74,11 +74,20 @@ def _pipe_1(
|
|
|
74
74
|
except AiModel.DoesNotExist:
|
|
75
75
|
logger.error(f"Pipe 1 failed: Model '{model_name}' not found.")
|
|
76
76
|
return False
|
|
77
|
+
|
|
77
78
|
except ModelMeta.DoesNotExist:
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
79
|
+
try:
|
|
80
|
+
model_name = download_segmentation_model()
|
|
81
|
+
ai_model_obj = AiModel.objects.get(name=model_name)
|
|
82
|
+
if model_meta_version is not None:
|
|
83
|
+
model_meta = ai_model_obj.metadata_versions.get(version=model_meta_version)
|
|
84
|
+
else:
|
|
85
|
+
model_meta = ModelMeta.setup_default_from_huggingface()
|
|
86
|
+
except ModelMeta.DoesNotExist:
|
|
87
|
+
logger.error(
|
|
88
|
+
f"Pipe 1 failed: ModelMeta version {model_meta_version} for model '{model_name}' not found."
|
|
89
|
+
)
|
|
90
|
+
return False
|
|
82
91
|
try:
|
|
83
92
|
sequences: Optional[Dict[str, List[Tuple[int, int]]]] = video_file.predict_video(
|
|
84
93
|
model_meta=model_meta,
|
|
@@ -18,7 +18,6 @@ from . import model_meta_logic as logic
|
|
|
18
18
|
|
|
19
19
|
if TYPE_CHECKING:
|
|
20
20
|
from endoreg_db.models import LabelSet, AiModel # pylint: disable=import-outside-toplevel
|
|
21
|
-
from torch.nn import Module as TorchModule
|
|
22
21
|
|
|
23
22
|
|
|
24
23
|
class ModelMetaManager(models.Manager):
|
|
@@ -132,15 +131,15 @@ class ModelMeta(models.Model):
|
|
|
132
131
|
@classmethod
|
|
133
132
|
def setup_default_from_huggingface(
|
|
134
133
|
cls: Type["ModelMeta"],
|
|
135
|
-
model_id: str,
|
|
136
|
-
task: str = "image-classification",
|
|
134
|
+
model_id: str = "wg-lux/colo_segmentation_RegNetX800MF_base",
|
|
137
135
|
labelset_name: Optional[str] = None,
|
|
138
136
|
) -> "ModelMeta":
|
|
139
137
|
"""
|
|
140
138
|
Downloads a pretrained model from Hugging Face and initializes ModelMeta automatically.
|
|
141
139
|
"""
|
|
142
|
-
|
|
143
|
-
return logic.setup_default_from_huggingface_logic(cls, model_id,
|
|
140
|
+
# If labelset_name is not provided, handle default logic here if needed
|
|
141
|
+
return logic.setup_default_from_huggingface_logic(cls, model_id, labelset_name)
|
|
142
|
+
|
|
144
143
|
|
|
145
144
|
|
|
146
145
|
@classmethod
|
|
@@ -166,12 +166,12 @@ class VideoImportService():
|
|
|
166
166
|
return None
|
|
167
167
|
raise
|
|
168
168
|
|
|
169
|
-
# Create sensitive meta file, ensure raw is moved out of processing folder watched by file watcher.
|
|
170
|
-
self._create_sensitive_file()
|
|
171
|
-
|
|
172
169
|
# Create or retrieve video instance
|
|
173
170
|
self._create_or_retrieve_video_instance()
|
|
174
171
|
|
|
172
|
+
# Create sensitive meta file, ensure raw is moved out of processing folder watched by file watcher.
|
|
173
|
+
self._create_sensitive_file()
|
|
174
|
+
|
|
175
175
|
# Setup processing environment
|
|
176
176
|
self._setup_processing_environment()
|
|
177
177
|
|
|
@@ -244,7 +244,6 @@ class VideoImportService():
|
|
|
244
244
|
|
|
245
245
|
def _create_or_retrieve_video_instance(self):
|
|
246
246
|
"""Create or retrieve the VideoFile instance and move to final storage."""
|
|
247
|
-
# Removed duplicate import of VideoFile (already imported at module level)
|
|
248
247
|
|
|
249
248
|
self.logger.info("Creating VideoFile instance...")
|
|
250
249
|
|
|
@@ -384,7 +384,7 @@ endoreg_db/models/media/pdf/report_reader/report_reader_config.py,sha256=wYVDmPS
|
|
|
384
384
|
endoreg_db/models/media/pdf/report_reader/report_reader_flag.py,sha256=j9tjbLRenxpWfeaseALl8rV2Dqem9YaM_duS1iJkARU,536
|
|
385
385
|
endoreg_db/models/media/video/__init__.py,sha256=ifW4SXXN2q6wAuFwSP7XlYskpX7UX6uy0py5mpCCOCM,211
|
|
386
386
|
endoreg_db/models/media/video/create_from_file.py,sha256=3n4bbzFteEOFDUuEikP0x-StCKI5R5IhyKC7o3kLZ6Y,15128
|
|
387
|
-
endoreg_db/models/media/video/pipe_1.py,sha256=
|
|
387
|
+
endoreg_db/models/media/video/pipe_1.py,sha256=yUzTi0pkw2ISsOoFpLmNky6S_V-TEWMxXmPLfB7gUpA,9899
|
|
388
388
|
endoreg_db/models/media/video/pipe_2.py,sha256=DnMxW0uOqSsf7-0n9Rlvn7u89U4Jpkv7n6hFpQfUjkQ,4964
|
|
389
389
|
endoreg_db/models/media/video/refactor_plan.md,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
390
390
|
endoreg_db/models/media/video/video_file.py,sha256=txlxR8d1OBgt3UEkWvLcGSyLarh0jXLw-z0SAV5KOok,26789
|
|
@@ -461,7 +461,7 @@ endoreg_db/models/medical/risk/risk.py,sha256=g5pgAfCfsvH88nbmX3xsASF3OZgNA-G6NJ
|
|
|
461
461
|
endoreg_db/models/medical/risk/risk_type.py,sha256=kEugcaWSTEWH_Vxq4dcF80Iv1L4_Kk1JKJGQMgz_s0o,1350
|
|
462
462
|
endoreg_db/models/metadata/__init__.py,sha256=8I6oLj3YTmeaPGJpL0AWG5gLwp38QzrEggxSkTisv7c,474
|
|
463
463
|
endoreg_db/models/metadata/frame_ocr_result.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
464
|
-
endoreg_db/models/metadata/model_meta.py,sha256=
|
|
464
|
+
endoreg_db/models/metadata/model_meta.py,sha256=F_r-PTLeNi4J-4EaGCQkGIguhdl7Bwba7_i56ZAjc-4,7589
|
|
465
465
|
endoreg_db/models/metadata/model_meta_logic.py,sha256=27mqScxUTJXNUVc6CqAs5dXjspEsh0TWPmlxdJVulGc,12015
|
|
466
466
|
endoreg_db/models/metadata/pdf_meta.py,sha256=BTmpSgqxmPKi0apcNjyrZAS4AFKCPXVdBd6VBeyyv6E,3174
|
|
467
467
|
endoreg_db/models/metadata/sensitive_meta.py,sha256=ekLHrW-b5uYcjfkRd0EW5ncx5ef8Bu-K6msDkpWCAbk,13034
|
|
@@ -600,7 +600,7 @@ endoreg_db/services/pseudonym_service.py,sha256=CJhbtRa6K6SPbphgCZgEMi8AFQtB18CU
|
|
|
600
600
|
endoreg_db/services/requirements_object.py,sha256=290zf8AEbVtCoHhW4Jr7_ud-RvrqYmb1Nz9UBHtTnc0,6164
|
|
601
601
|
endoreg_db/services/segment_sync.py,sha256=YgHvIHkbW4mqCu0ACf3zjRSZnNfxWwt4gh5syUVXuE0,6400
|
|
602
602
|
endoreg_db/services/storage_aware_video_processor.py,sha256=kKFK64vXLeBSVkp1YJonU3gFDTeXZ8C4qb9QZZB99SE,13420
|
|
603
|
-
endoreg_db/services/video_import.py,sha256=
|
|
603
|
+
endoreg_db/services/video_import.py,sha256=Ifl-x1WSlHEcA-Lzf75l_b84g8LqXXUA_OmENZhjv3A,47747
|
|
604
604
|
endoreg_db/tasks/upload_tasks.py,sha256=OJq7DhNwcbWdXzHY8jz5c51BCVkPN5gSWOz-6Fx6W5M,7799
|
|
605
605
|
endoreg_db/tasks/video_ingest.py,sha256=kxFuYkHijINV0VabQKCFVpJRv6eCAw07tviONurDgg8,5265
|
|
606
606
|
endoreg_db/tasks/video_processing_tasks.py,sha256=KjcERRJ1TZzmavBpvr6OsvSTUViU0PR1ECWnEdzu2Js,14140
|
|
@@ -784,7 +784,7 @@ endoreg_db/views/video/video_meta.py,sha256=C1wBMTtQb_yzEUrhFGAy2UHEWMk_CbU75WXX
|
|
|
784
784
|
endoreg_db/views/video/video_processing_history.py,sha256=mhFuS8RG5GV8E-lTtuD0qrq-bIpnUFp8vy9aERfC-J8,770
|
|
785
785
|
endoreg_db/views/video/video_remove_frames.py,sha256=2FmvNrSPM0fUXiBxINN6vBUUDCqDlBkNcGR3WsLDgKo,1696
|
|
786
786
|
endoreg_db/views/video/video_stream.py,sha256=kLyuf0ORTmsLeYUQkTQ6iRYqlIQozWhMMR3Lhfe_trk,12148
|
|
787
|
-
endoreg_db-0.8.
|
|
788
|
-
endoreg_db-0.8.
|
|
789
|
-
endoreg_db-0.8.
|
|
790
|
-
endoreg_db-0.8.
|
|
787
|
+
endoreg_db-0.8.3.1.dist-info/METADATA,sha256=M6P6tLtoK5aa7AEUO9ZjJxAIe96STW10oY4grrihNYU,14758
|
|
788
|
+
endoreg_db-0.8.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
789
|
+
endoreg_db-0.8.3.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
|
|
790
|
+
endoreg_db-0.8.3.1.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|