mnist-ready 1.1.2 → 1.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7696501a6c74718d41b63494e633f6f6b73be2372956fa9cb1d0fd084d4837a3
-  data.tar.gz: 94261b6706c88f049359a1d88ee667fd75b5df407c93e10d9ee5a6277d0437e4
+  metadata.gz: 4e7b27f81d9d8aa860901b53d849477e1c204fe97f9ee18811caba4fa3b1bf02
+  data.tar.gz: 7de49c1978d0457021e28be6f4906685af00ca18bd90cdc5c37d637d32f61445
 SHA512:
-  metadata.gz: 0bc784b1a75a6cdb9be706d2d5017f670cabc8ab371205d4d2d84f45c145e74269503a29f61c8371c1b86b4184b7308d443c23c564fb1d1d7664633d3722115a
-  data.tar.gz: a889be91d16e0eed699f5ec1298c13a3225ea355a074cc44d0c057776bc696040f790494d18f1643f3d5cd1a2b8591784680486fcde88bc5bdc4a73fac023aef
+  metadata.gz: 84a37ee74b5d893fd3e0f4fce0eb1e90deb7b925e4a6b89094aa3ea9f1ba1f0610f9139ef7d57e7cf5be62e40a5a054164751443d3476db27d81cfb0bf00b108
+  data.tar.gz: 2c7882315bff945712be2c85c3a820fe11fdfaad47c132c8c2a130a350031bab6a05e6d99410cec36b5f71ce20fb43303b2a83d9efd8f4a3cf0eff30ef6935bf

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.3
 platform: ruby
 authors:
 - Sergio Oliveira Jr
@@ -9,7 +9,21 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2024-09-19 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rmagick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
 description:
 email: sergio.oliveira.jr@gmail.com
 executables: []
@@ -20,6 +34,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