From 5f2b04e834937ee07ea2240534c2eaad229f4276 Mon Sep 17 00:00:00 2001
From: TimNiklasWitte <wittet@uos.de>
Date: Thu, 31 Mar 2022 14:26:00 +0200
Subject: [PATCH] add comment that code for loading imagenet dataset copied
 from blog article

---
 Training.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Training.py b/Training.py
index 9a2d0f7..9e4cee8 100644
--- a/Training.py
+++ b/Training.py
@@ -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())