Update model_definition.py

model_definition.py

@@ -13,7 +13,27 @@ from torchvision import transforms
 from functools import partial
 from typing import Optional, List
 from torch import Tensor
-
+import os
+import json
+import numpy as np
+import cv2
+from PIL import Image
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+from torchvision import transforms
+from functools import partial
+from collections import deque, OrderedDict
+import math
+from torch.nn import MultiheadAttention
+from torch.nn import TransformerEncoder, TransformerEncoderLayer
+from torch.nn import TransformerDecoder, TransformerDecoderLayer
+from timm.models.resnet import resnet50d, resnet26d, resnet18d
+try:
+    from timm.layers import trunc_normal_
+except ImportError:
+    from timm.models.layers import trunc_normal_
+from scipy.ndimage import maximum_filter
 # مكتبات إضافية
 import os
 import json