feat(tf): add birds classifier

stakach · Jun 5, 2022 · 933e880 · 933e880
1 parent f21fbce
commit 933e880
Show file tree

Hide file tree

Showing 5 changed files with 3,872 additions and 1 deletion.
diff --git a/src/kritter/tf/__init__.py b/src/kritter/tf/__init__.py
@@ -14,4 +14,5 @@
 
 from .tfdetector import TFDetector
 COCO = os.path.join(_basepath, "coco") 
-BIRDFEEDER = os.path.join(_basepath, "birdfeeder") 
+BIRDFEEDER = os.path.join(_basepath, "birdfeeder") 
+BIRDS = os.path.join(_basepath, "birds") 
diff --git a/src/kritter/tf/birds/convert_graph.cr b/src/kritter/tf/birds/convert_graph.cr
@@ -0,0 +1,11 @@
+require "csv"
+
+file_input = ARGV[0]
+file_output = ARGV[1]
+puts "Reading #{file_input}"
+puts "Writing #{file_output}"
+File.open(file_output, "w") do |output|
+  CSV.each_row(File.open(file_input)) do |row|
+    output << %(item {\n  id: #{row[0]}\n  name: '#{row[1]}'\n}\n)
+  end
+end
diff --git a/src/kritter/tf/birds/frozen_inference_graph.pb b/src/kritter/tf/birds/frozen_inference_graph.pb