mnist-ready 1.1.1 → 1.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b2b0059fb779cef3b208d6b061f44cf0276050bf2ab03c7ba35d26433bdb119
-  data.tar.gz: 964e0569bd60637da0c376d054358a9ed70512942f238231050533a1028dab40
+  metadata.gz: 4e7b27f81d9d8aa860901b53d849477e1c204fe97f9ee18811caba4fa3b1bf02
+  data.tar.gz: 7de49c1978d0457021e28be6f4906685af00ca18bd90cdc5c37d637d32f61445
 SHA512:
-  metadata.gz: bf6e5fda8c12fb6dac08c923db72c32b3948bfeb432c03310f88dd711a54d85b847a4c124b5d1787f98f115ff7186601b0c2472cd37c8a5fdbca3e1fc8ae8bab
-  data.tar.gz: 3622880f3bafe45b3b621221ebfa9c3f02459c9a76551898c38efdef3087cb1cc74b411bf0889e76b3f4081c01a95d2c714c73ebf52ea5eac1c96128fbf57904
+  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_dataset.rb

@@ -11,7 +11,7 @@ class MnistDataset
   
     private_class_method :new
   
-    def initialize(show_progress)
+    def initialize(show_progress, just_one)
         test_set = File.join(File.dirname(__FILE__), '..', 'data', 'mnist_test.csv')
 
         @train_set = []
@@ -24,6 +24,7 @@ class MnistDataset
             @test_set.push(digit)
             @all_set.push(digit)
             print "\r#{all_set.size}" if show_progress
+            break if just_one
         end
 
         (1..3).each do |index|
@@ -34,17 +35,19 @@ class MnistDataset
                 @train_set.push(digit)
                 @all_set.push(digit)
                 print "\r#{all_set.size}" if show_progress
+                break if just_one
             end
+            break if just_one
         end
         puts if show_progress
     end
 
     attr_reader :all_set, :train_set, :test_set
 
-    def self.instance(show_progress = false)
+    def self.instance(show_progress = false, just_one = false)
       return @instance if @instance
       @instance_mutex.synchronize do
-        @instance ||= new(show_progress)
+        @instance ||= new(show_progress, just_one)
       end
       @instance
     end
@@ -97,13 +100,12 @@ class MnistDataset
 
     def count_digits(hash) 
         count = {}
+        (0..9).each do |label|
+            count[label] = 0
+        end
         hash.each do |digit|
-            label = digit.label
-            if count.has_key?(label)
-                count[label] = count[label] + 1
-            else
-                count[label] = 1
-            end
+          label = digit.label
+          count[label] = count[label] + 1
         end
         count
     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,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: mnist-ready
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.1.3
 platform: ruby
 authors:
 - Sergio Oliveira Jr
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-02 00:00:00.000000000 Z
-dependencies: []
+date: 2024-09-19 00:00:00.000000000 Z
+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
@@ -42,7 +57,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.7
+rubygems_version: 3.4.22
 signing_key:
 specification_version: 4
 summary: A simple and straightforward Ruby library that handles everything related