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

add comment that code for loading imagenet dataset copied from blog article

Browse Source
This commit is contained in:
TimNiklasWitte
2022-03-31 14:26:00 +02:00
parent 6913e305eb
commit 5f2b04e834
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Training.py
View File

@@ -29,6 +29,8 @@ def getRGB(L, AB, batch_mode=True):
def main():
# Code for loading the ImageNet2012 dataset taken from:
# https://medium.com/analytics-vidhya/how-to-train-a-neural-network-classifier-on-imagenet-using-tensorflow-2-ede0ea3a35ff
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())