administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-29 14:20:49 -06:00
parent 93ccd7f311
commit 4066201ad3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/util/classification.py
View File

@ -71,7 +71,7 @@ def train_classification_model(model_dir: str) -> bool:
# create training set # create training set
datagen = ImageDataGenerator(rescale=1.0 / 255, validation_split=0.2) datagen = ImageDataGenerator(rescale=1.0 / 255, validation_split=0.2)
train_gen = datagen.flow_from_directory( train_gen = datagen.flow_from_directory(
"train", dataset_dir,
target_size=(224, 224), target_size=(224, 224),
batch_size=BATCH_SIZE, batch_size=BATCH_SIZE,
class_mode="categorical", class_mode="categorical",