From e31ae4c47c08c731039562fdde93fc76e07c3768 Mon Sep 17 00:00:00 2001
From: hitharevalla <43685030+hitharevalla@users.noreply.github.com>
Date: Mon, 13 May 2019 13:37:49 -0700
Subject: [PATCH] Update generate_tfrecord.py

---
 generate_tfrecord.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generate_tfrecord.py b/generate_tfrecord.py
index 87fa5a98..0ad13386 100644
--- a/generate_tfrecord.py
+++ b/generate_tfrecord.py
@@ -29,7 +29,7 @@
 
 # TO-DO replace this with label map
 def class_text_to_int(row_label):
-    if row_label == 'raccoon':
+    if row_label == 'hand':
         return 1
     else:
         None