lib/dnn/lib/image_io.rb in ruby-dnn-0.6.3 vs lib/dnn/lib/image_io.rb in ruby-dnn-0.6.4

@@ -18,15 +18,15 @@
         img = Numo::UInt8[img, img, img].transpose(1, 2, 0).clone
       end
       h, w, ch = img.shape
       bin = img.to_binary
       case file_name
-      when /\.png$/
+      when /\.png$/i
         stride_in_bytes = w * ch
         Stb.stbi_write_png(file_name, w, h, ch, bin, stride_in_bytes)
-      when /\.bmp$/
+      when /\.bmp$/i
         Stb.stbi_write_bmp(file_name, w, h, ch, bin)
-      when /\.jpg$/
+      when /\.jpg$/i, /\.jpeg/i
         Stb.stbi_write_jpg(file_name, w, h, ch, bin, quality)
       end
     rescue => ex
       raise ImageIO::WriteError.new(ex.message)
     end