mnist-ready 1.1.2 → 1.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7696501a6c74718d41b63494e633f6f6b73be2372956fa9cb1d0fd084d4837a3
-  data.tar.gz: 94261b6706c88f049359a1d88ee667fd75b5df407c93e10d9ee5a6277d0437e4
+  metadata.gz: d87018c07bc7327b25f7b687afece352a63a13d8d050d2a6301c384dddb4aa4d
+  data.tar.gz: 8d8666a208cc052a0c35688775aaa47a1d2307f2a395880442cff04160e5f4e0
 SHA512:
-  metadata.gz: 0bc784b1a75a6cdb9be706d2d5017f670cabc8ab371205d4d2d84f45c145e74269503a29f61c8371c1b86b4184b7308d443c23c564fb1d1d7664633d3722115a
-  data.tar.gz: a889be91d16e0eed699f5ec1298c13a3225ea355a074cc44d0c057776bc696040f790494d18f1643f3d5cd1a2b8591784680486fcde88bc5bdc4a73fac023aef
+  metadata.gz: 000af885e1987f73875c140a9e0589969914b7153b7e985baf1374ffb870ab011c11d712424ff358a642558c1007eb07668469f0ec08d45133429b998eb4c5c2
+  data.tar.gz: 94ee6a7e5fd39674b77db13ed4fb48aac5a3b1ac21cf5618cbbce1a8df57cc72a62b0f3a561f816ca5418cf30ba9a49a1e871b253d3efa9d180bf145d0adc315

data/lib/draw_image.rb

@@ -0,0 +1,41 @@
+require 'rmagick'
+require 'base64'
+
+class DrawImage
+
+    def self.draw(pixel_array, white_background = false)
+
+        # Create a new 28x28 image
+        image = Magick::Image.new(28, 28)
+
+        # Set each pixel in the image
+        28.times do |y|
+            28.times do |x|
+                pixel_value = pixel_array[y * 28 + x]
+                pixel_value = 255 - pixel_value if white_background
+                color = Magick::Pixel.new(pixel_value * 257, pixel_value * 257, pixel_value * 257)
+                image.pixel_color(x, y, color)
+            end
+        end
+
+        # Scale up the image to make it more visible
+        image = image.scale(10)
+
+        # Set the format to PNG
+        image.format = 'PNG'
+
+        # Convert the image to a blob
+        blob = image.to_blob
+
+        # Encode the blob as base64
+        encoded_image = Base64.strict_encode64(blob)
+
+        # Display the image inline using iTerm2's imgcat escape sequence
+        puts "\033]1337;File=inline=1;size=#{blob.size}:#{encoded_image}\007"
+        
+        # Force a newline after the image
+        puts "\n"
+
+    end
+
+end

data/lib/mnist_digit.rb

@@ -13,6 +13,12 @@ class MnistDigit
     def pixels
       @pixels
     end
+
+    def png_image(white_background = false)
+      require_relative 'draw_image.rb'
+      puts
+      DrawImage.draw(pixels, white_background)
+    end
   
     def ascii_image
       img = ""

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mnist-ready
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.1.4
 platform: ruby
 authors:
 - Sergio Oliveira Jr
@@ -20,6 +20,7 @@ files:
 - data/mnist_train1.csv
 - data/mnist_train2.csv
 - data/mnist_train3.csv
+- lib/draw_image.rb
 - lib/mnist-ready.rb
 - lib/mnist_dataset.rb
 - lib/mnist_digit.rb