qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-06-27 23:39:15 +02:00
Code Issues Projects Releases Wiki Activity

training.metrics: allow module init without TFA

Browse source
This commit is contained in:
Robert Sachunsky 2026-06-19 22:01:02 +02:00
parent eb4cae9dee
commit 0bfbbfdc80
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/training/metrics.py
View file

@ -5,7 +5,10 @@ import tensorflow as tf
from tensorflow.keras import backend as K
from tensorflow.keras.metrics import Metric, MeanMetricWrapper, get
from tensorflow.keras.initializers import Zeros
try:
from tensorflow_addons.image import connected_components
except ModuleNotFoundError:
pass # n/a beyond TF 2.15 (and only needed for training) ...
import numpy as np