torchvision 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 19fff61bb461e5fbf850702bf485a442b2013807ac28549bc427e5f3d8c7472b
-  data.tar.gz: bc8004d2ca26e9022f2fa2b1663277bcedf701729f59aae248058a26e605a5ad
+  metadata.gz: c6ae301007e3310dfc378a7f76d24bd71cc048be6156b32d9e0705412817c565
+  data.tar.gz: 423aee54316683ddf9d3893e04f8836dd16ade1408f6c046f6d8f6f337a09a46
 SHA512:
-  metadata.gz: fbd3d7292efa6ee2fd2c0ff8cb85659d37a19761b4d93a9a4923a9990d7400c849738913db12720e7d232d6fdb180c16f06da0ecc3601a922bf0036beb0b44bd
-  data.tar.gz: '09b86d6b01f25d43ac65d9c3d0509b3488ed2108d57090553ae46526f841551cbdd16fefce1cb15ea7f326d946e3abb26d777c3c45e23538f8fb7753fdb6fec9'
+  metadata.gz: cf9c5a514c2ef42299161f97f077897c11b4708858018ffefcb282d6434003d49555ba54ea93ef0fb9e2f3cafdd7bd4395677e02bdfa5b1dcad4772aaeccc896
+  data.tar.gz: 0e01311267e620be6dfbed77724baed42ad9c597a2b5452d762845597ae257932e76848e0069149598743d78dc08b0374b18897ce9a272640eed4a3cec21bbd9

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.1 (2020-04-28)
+
+- Removed `mini_magick` for performance
+
 ## 0.1.0 (2020-04-27)
 
 - First release

data/README.md

@@ -4,6 +4,8 @@
 
 This gem is currently experimental. There may be breaking changes between each release. Please report any issues you experience.
 
+[![Build Status](https://travis-ci.org/ankane/torchvision.svg?branch=master)](https://travis-ci.org/ankane/torchvision)
+
 ## Installation
 
 Add this line to your application’s Gemfile:

data/lib/torchvision.rb

@@ -1,5 +1,4 @@
 # dependencies
-require "mini_magick"
 require "numo/narray"
 require "torch"
 
@@ -7,6 +6,7 @@ require "torch"
 require "digest"
 require "fileutils"
 require "net/http"
+require "tmpdir"
 
 # modules
 require "torchvision/version"

data/lib/torchvision/datasets/mnist.rb

@@ -43,7 +43,6 @@ module TorchVision
 
       def [](index)
         img = @data[index]
-        img = MiniMagick::Image.import_pixels(img.numo.to_binary, img.size(0), img.size(1), 8, "gray")
         img = @transform.call(img) if @transform
 
         target = @targets[index].item

data/lib/torchvision/transforms/functional.rb

@@ -34,7 +34,8 @@ module TorchVision
 
         # TODO improve
         def to_tensor(pic)
-          Torch.tensor(pic.get_pixels, dtype: :float)
+          pic = pic.float
+          pic.unsqueeze!(0).div!(255)
         end
       end
     end

data/lib/torchvision/version.rb

@@ -1,3 +1,3 @@
 module TorchVision
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: torchvision
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-04-27 00:00:00.000000000 Z
+date: 2020-04-28 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: mini_magick
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 - !ruby/object:Gem::Dependency
   name: numo-narray
   requirement: !ruby/object:Gem::Requirement
@@ -44,14 +30,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.2
+        version: 0.2.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.2
+        version: 0.2.3
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement