Update

Author: rybakov-ks

Images/8.png

@@ -1,15 +1,12 @@
 from sahi.models.detectron2 import Detectron2DetectionModel
-from detectron2.config import get_cfg
-import os
 
 """Переопределяем load_model для скрытия ошибки"""
 class CustomDetectron2Model(Detectron2DetectionModel):
     def load_model(self):
         from detectron2.config import get_cfg
         from detectron2.data import MetadataCatalog
         from detectron2.engine import DefaultPredictor
-        from detectron2.model_zoo import model_zoo
-
+        
         cfg = get_cfg()
 
         try:  # Переделываем импорт конфига и модели для скрытия ошибки "{} not available in Model Zoo!"

particleanalyzer/core/CustomDetectron2Model.py

@@ -9,7 +9,6 @@
 from tqdm import tqdm
 from typing import Optional, Tuple, List, Dict, Any, Union
 from scipy.spatial.distance import pdist
-
 try:
     from detectron2.engine import DefaultPredictor
     from particleanalyzer.core.CustomDetectron2Model import CustomDetectron2Model
@@ -136,7 +135,6 @@ def analyze_image(self, image: np.ndarray, image2: Optional[np.ndarray],
 
             # Выбор стратегии обработки
             processor = self._select_processor(model_change, sahi_mode)
-
             output_image, particle_data, annotations = processor(image, 
                     scale_input, confidence_threshold,
                     scale_selector, confidence_iou,