torchvision 0.1.1 → 0.2.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (47) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +35 -0
  3. data/LICENSE.txt +1 -1
  4. data/README.md +133 -7
  5. data/lib/torchvision.rb +39 -0
  6. data/lib/torchvision/datasets/cifar10.rb +117 -0
  7. data/lib/torchvision/datasets/cifar100.rb +41 -0
  8. data/lib/torchvision/datasets/dataset_folder.rb +91 -0
  9. data/lib/torchvision/datasets/fashion_mnist.rb +30 -0
  10. data/lib/torchvision/datasets/image_folder.rb +12 -0
  11. data/lib/torchvision/datasets/kmnist.rb +30 -0
  12. data/lib/torchvision/datasets/mnist.rb +47 -75
  13. data/lib/torchvision/datasets/vision_dataset.rb +67 -0
  14. data/lib/torchvision/models/alexnet.rb +42 -0
  15. data/lib/torchvision/models/basic_block.rb +46 -0
  16. data/lib/torchvision/models/bottleneck.rb +47 -0
  17. data/lib/torchvision/models/resnet.rb +129 -0
  18. data/lib/torchvision/models/resnet101.rb +9 -0
  19. data/lib/torchvision/models/resnet152.rb +9 -0
  20. data/lib/torchvision/models/resnet18.rb +9 -0
  21. data/lib/torchvision/models/resnet34.rb +9 -0
  22. data/lib/torchvision/models/resnet50.rb +9 -0
  23. data/lib/torchvision/models/resnext101_32x8d.rb +11 -0
  24. data/lib/torchvision/models/resnext50_32x4d.rb +11 -0
  25. data/lib/torchvision/models/vgg.rb +93 -0
  26. data/lib/torchvision/models/vgg11.rb +9 -0
  27. data/lib/torchvision/models/vgg11_bn.rb +9 -0
  28. data/lib/torchvision/models/vgg13.rb +9 -0
  29. data/lib/torchvision/models/vgg13_bn.rb +9 -0
  30. data/lib/torchvision/models/vgg16.rb +9 -0
  31. data/lib/torchvision/models/vgg16_bn.rb +9 -0
  32. data/lib/torchvision/models/vgg19.rb +9 -0
  33. data/lib/torchvision/models/vgg19_bn.rb +9 -0
  34. data/lib/torchvision/models/wide_resnet101_2.rb +10 -0
  35. data/lib/torchvision/models/wide_resnet50_2.rb +10 -0
  36. data/lib/torchvision/transforms/center_crop.rb +13 -0
  37. data/lib/torchvision/transforms/compose.rb +2 -2
  38. data/lib/torchvision/transforms/functional.rb +142 -8
  39. data/lib/torchvision/transforms/normalize.rb +2 -2
  40. data/lib/torchvision/transforms/random_horizontal_flip.rb +18 -0
  41. data/lib/torchvision/transforms/random_resized_crop.rb +70 -0
  42. data/lib/torchvision/transforms/random_vertical_flip.rb +18 -0
  43. data/lib/torchvision/transforms/resize.rb +13 -0
  44. data/lib/torchvision/transforms/to_tensor.rb +2 -2
  45. data/lib/torchvision/utils.rb +118 -0
  46. data/lib/torchvision/version.rb +1 -1
  47. metadata +51 -44
data/lib/torchvision/version.rb CHANGED
@@ -1,3 +1,3 @@
1
1
  module TorchVision
2
- VERSION = "0.1.1"
2
+ VERSION = "0.2.2"
3
3
  end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: torchvision
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.1
4
+ version: 0.2.2
5
5
  platform: ruby
6
6
  authors:
7
7
  - Andrew Kane
8
- autorequire:
8
+ autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2020-04-28 00:00:00.000000000 Z
11
+ date: 2021-05-23 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: numo-narray
@@ -25,63 +25,35 @@ dependencies:
25
25
  - !ruby/object:Gem::Version
26
26
  version: '0'
27
27
  - !ruby/object:Gem::Dependency
28
- name: torch-rb
28
+ name: ruby-vips
29
29
  requirement: !ruby/object:Gem::Requirement
30
30
  requirements:
31
31
  - - ">="
32
32
  - !ruby/object:Gem::Version
33
- version: 0.2.3
33
+ version: 2.1.2
34
34
  type: :runtime
35
35
  prerelease: false
36
36
  version_requirements: !ruby/object:Gem::Requirement
37
37
  requirements:
38
38
  - - ">="
39
39
  - !ruby/object:Gem::Version
40
- version: 0.2.3
41
- - !ruby/object:Gem::Dependency
42
- name: bundler
43
- requirement: !ruby/object:Gem::Requirement
44
- requirements:
45
- - - ">="
46
- - !ruby/object:Gem::Version
47
- version: '0'
48
- type: :development
49
- prerelease: false
50
- version_requirements: !ruby/object:Gem::Requirement
51
- requirements:
52
- - - ">="
53
- - !ruby/object:Gem::Version
54
- version: '0'
40
+ version: 2.1.2
55
41
  - !ruby/object:Gem::Dependency
56
- name: rake
57
- requirement: !ruby/object:Gem::Requirement
58
- requirements:
59
- - - ">="
60
- - !ruby/object:Gem::Version
61
- version: '0'
62
- type: :development
63
- prerelease: false
64
- version_requirements: !ruby/object:Gem::Requirement
65
- requirements:
66
- - - ">="
67
- - !ruby/object:Gem::Version
68
- version: '0'
69
- - !ruby/object:Gem::Dependency
70
- name: minitest
42
+ name: torch-rb
71
43
  requirement: !ruby/object:Gem::Requirement
72
44
  requirements:
73
45
  - - ">="
74
46
  - !ruby/object:Gem::Version
75
- version: '5'
76
- type: :development
47
+ version: 0.3.7
48
+ type: :runtime
77
49
  prerelease: false
78
50
  version_requirements: !ruby/object:Gem::Requirement
79
51
  requirements:
80
52
  - - ">="
81
53
  - !ruby/object:Gem::Version
82
- version: '5'
83
- description:
84
- email: andrew@chartkick.com
54
+ version: 0.3.7
55
+ description:
56
+ email: andrew@ankane.org
85
57
  executables: []
86
58
  extensions: []
87
59
  extra_rdoc_files: []
@@ -90,17 +62,52 @@ files:
90
62
  - LICENSE.txt
91
63
  - README.md
92
64
  - lib/torchvision.rb
65
+ - lib/torchvision/datasets/cifar10.rb
66
+ - lib/torchvision/datasets/cifar100.rb
67
+ - lib/torchvision/datasets/dataset_folder.rb
68
+ - lib/torchvision/datasets/fashion_mnist.rb
69
+ - lib/torchvision/datasets/image_folder.rb
70
+ - lib/torchvision/datasets/kmnist.rb
93
71
  - lib/torchvision/datasets/mnist.rb
72
+ - lib/torchvision/datasets/vision_dataset.rb
73
+ - lib/torchvision/models/alexnet.rb
74
+ - lib/torchvision/models/basic_block.rb
75
+ - lib/torchvision/models/bottleneck.rb
76
+ - lib/torchvision/models/resnet.rb
77
+ - lib/torchvision/models/resnet101.rb
78
+ - lib/torchvision/models/resnet152.rb
79
+ - lib/torchvision/models/resnet18.rb
80
+ - lib/torchvision/models/resnet34.rb
81
+ - lib/torchvision/models/resnet50.rb
82
+ - lib/torchvision/models/resnext101_32x8d.rb
83
+ - lib/torchvision/models/resnext50_32x4d.rb
84
+ - lib/torchvision/models/vgg.rb
85
+ - lib/torchvision/models/vgg11.rb
86
+ - lib/torchvision/models/vgg11_bn.rb
87
+ - lib/torchvision/models/vgg13.rb
88
+ - lib/torchvision/models/vgg13_bn.rb
89
+ - lib/torchvision/models/vgg16.rb
90
+ - lib/torchvision/models/vgg16_bn.rb
91
+ - lib/torchvision/models/vgg19.rb
92
+ - lib/torchvision/models/vgg19_bn.rb
93
+ - lib/torchvision/models/wide_resnet101_2.rb
94
+ - lib/torchvision/models/wide_resnet50_2.rb
95
+ - lib/torchvision/transforms/center_crop.rb
94
96
  - lib/torchvision/transforms/compose.rb
95
97
  - lib/torchvision/transforms/functional.rb
96
98
  - lib/torchvision/transforms/normalize.rb
99
+ - lib/torchvision/transforms/random_horizontal_flip.rb
100
+ - lib/torchvision/transforms/random_resized_crop.rb
101
+ - lib/torchvision/transforms/random_vertical_flip.rb
102
+ - lib/torchvision/transforms/resize.rb
97
103
  - lib/torchvision/transforms/to_tensor.rb
104
+ - lib/torchvision/utils.rb
98
105
  - lib/torchvision/version.rb
99
106
  homepage: https://github.com/ankane/torchvision
100
107
  licenses:
101
108
  - BSD-3-Clause
102
109
  metadata: {}
103
- post_install_message:
110
+ post_install_message:
104
111
  rdoc_options: []
105
112
  require_paths:
106
113
  - lib
@@ -108,15 +115,15 @@ required_ruby_version: !ruby/object:Gem::Requirement
108
115
  requirements:
109
116
  - - ">="
110
117
  - !ruby/object:Gem::Version
111
- version: '2.4'
118
+ version: '2.6'
112
119
  required_rubygems_version: !ruby/object:Gem::Requirement
113
120
  requirements:
114
121
  - - ">="
115
122
  - !ruby/object:Gem::Version
116
123
  version: '0'
117
124
  requirements: []
118
- rubygems_version: 3.1.2
119
- signing_key:
125
+ rubygems_version: 3.2.3
126
+ signing_key:
120
127
  specification_version: 4
121
128
  summary: Computer vision datasets, transforms, and models for Ruby
122
129
  test_files: []