qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-06 22:50:14 +02:00
Code Issues Projects Releases Wiki Activity

remove unused imports

Browse source
This commit is contained in:
kba 2025-10-01 18:10:13 +02:00
parent 95bb5908bb
commit f0ef2b5db2
2 changed files with 1 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
train/gt_gen_utils.py
View file

@ -1,5 +1,3 @@
import click
import sys
import os import os
import numpy as np import numpy as np
import warnings import warnings
@ -9,7 +7,7 @@ import cv2
from shapely import geometry from shapely import geometry
from pathlib import Path from pathlib import Path
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont from PIL import ImageFont
KERNEL = np.ones((5, 5), np.uint8) KERNEL = np.ones((5, 5), np.uint8)

6
train/inference.py
View file

@ -3,12 +3,9 @@ import os
import numpy as np import numpy as np
import warnings import warnings
import cv2 import cv2
import seaborn as sns
from tensorflow.keras.models import load_model from tensorflow.keras.models import load_model
import tensorflow as tf import tensorflow as tf
from tensorflow.keras import backend as K from tensorflow.keras import backend as K
from tensorflow.keras import layers
import tensorflow.keras.losses
from tensorflow.keras.layers import * from tensorflow.keras.layers import *
from models import * from models import *
from gt_gen_utils import * from gt_gen_utils import *
@ -16,7 +13,6 @@ import click
import json import json
from tensorflow.python.keras import backend as tensorflow_backend from tensorflow.python.keras import backend as tensorflow_backend
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
import matplotlib.pyplot as plt
with warnings.catch_warnings(): with warnings.catch_warnings():
@ -55,11 +51,9 @@ class sbb_predict:
seg=seg[:,:,0] seg=seg[:,:,0]
seg_img=np.zeros((np.shape(seg)[0],np.shape(seg)[1],3)).astype(np.uint8) seg_img=np.zeros((np.shape(seg)[0],np.shape(seg)[1],3)).astype(np.uint8)
colors=sns.color_palette("hls", self.n_classes)
for c in ann_u: for c in ann_u:
c=int(c) c=int(c)
segl=(seg==c)
seg_img[:,:,0][seg==c]=c seg_img[:,:,0][seg==c]=c
seg_img[:,:,1][seg==c]=c seg_img[:,:,1][seg==c]=c
seg_img[:,:,2][seg==c]=c seg_img[:,:,2][seg==c]=c