16 lines
343 B
Python
16 lines
343 B
Python
|
import tensorflow as tf
|
||
|
|
|
||
|
|
ds_train, ds_valid = tf.keras.preprocessing.image_dataset_from_directory(
|
||
|
|
'../dataset-orig',
|
||
|
|
labels = 'inferred',
|
||
|
|
label_mode = 'categorical',
|
||
|
|
color_mode = 'rgb',
|
||
|
|
image_size = (300, 300),
|
||
|
|
shuffle = False,
|
||
|
|
validation_split = 0.05,
|
||
|
|
subset = 'both',
|
||
|
|
verbose = True
|
||
|
|
)
|
||
|
|
|
||
|
|
from m import mod
|