denshooter/gpu_colorization
1
0
Fork 0
mirror of https://github.com/denshooter/gpu_colorization.git synced 2026-01-21 12:32:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

expand readme

Browse Source
This commit is contained in:
TimNiklasWitte
2022-03-31 14:22:47 +02:00
parent 20cab55ebb
commit 9a5405a6df
2 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Training.py
View File

@@ -33,7 +33,7 @@ def main():
labels_path = tf.keras.utils.get_file('ImageNetLabels.txt','https://storage.googleapis.com/download.tensorflow.org/data/ImageNetLabels.txt')
imagenet_labels = np.array(open(labels_path).read().splitlines())
data_dir = '/home/timwitte/Downloads/'
data_dir = '/home/timwitte/Downloads/' # <---- change me!!!!
write_dir = './imagenet'
# Construct a tf.data.Dataset