torchvision 0.1.0 → 0.2.1

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 -5
  5. data/lib/torchvision.rb +40 -1
  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 -76
  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 -7
  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 +120 -0
  46. data/lib/torchvision/version.rb +1 -1
  47. metadata +50 -57
data/lib/torchvision/version.rb CHANGED
@@ -1,3 +1,3 @@
1
1
  module TorchVision
2
- VERSION = "0.1.0"
2
+ VERSION = "0.2.1"
3
3
  end
metadata CHANGED
@@ -1,17 +1,17 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: torchvision
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.0
4
+ version: 0.2.1
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-27 00:00:00.000000000 Z
11
+ date: 2021-03-15 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
- name: mini_magick
14
+ name: numo-narray
15
15
  requirement: !ruby/object:Gem::Requirement
16
16
  requirements:
17
17
  - - ">="
@@ -25,77 +25,35 @@ dependencies:
25
25
  - !ruby/object:Gem::Version
26
26
  version: '0'
27
27
  - !ruby/object:Gem::Dependency
28
- name: numo-narray
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'
33
+ version: '2.1'
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'
40
+ version: '2.1'
41
41
  - !ruby/object:Gem::Dependency
42
42
  name: torch-rb
43
43
  requirement: !ruby/object:Gem::Requirement
44
44
  requirements:
45
45
  - - ">="
46
46
  - !ruby/object:Gem::Version
47
- version: 0.2.2
47
+ version: 0.3.7
48
48
  type: :runtime
49
49
  prerelease: false
50
50
  version_requirements: !ruby/object:Gem::Requirement
51
51
  requirements:
52
52
  - - ">="
53
53
  - !ruby/object:Gem::Version
54
- version: 0.2.2
55
- - !ruby/object:Gem::Dependency
56
- name: bundler
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: rake
71
- requirement: !ruby/object:Gem::Requirement
72
- requirements:
73
- - - ">="
74
- - !ruby/object:Gem::Version
75
- version: '0'
76
- type: :development
77
- prerelease: false
78
- version_requirements: !ruby/object:Gem::Requirement
79
- requirements:
80
- - - ">="
81
- - !ruby/object:Gem::Version
82
- version: '0'
83
- - !ruby/object:Gem::Dependency
84
- name: minitest
85
- requirement: !ruby/object:Gem::Requirement
86
- requirements:
87
- - - ">="
88
- - !ruby/object:Gem::Version
89
- version: '5'
90
- type: :development
91
- prerelease: false
92
- version_requirements: !ruby/object:Gem::Requirement
93
- requirements:
94
- - - ">="
95
- - !ruby/object:Gem::Version
96
- version: '5'
97
- description:
98
- email: andrew@chartkick.com
54
+ version: 0.3.7
55
+ description:
56
+ email: andrew@ankane.org
99
57
  executables: []
100
58
  extensions: []
101
59
  extra_rdoc_files: []
@@ -104,17 +62,52 @@ files:
104
62
  - LICENSE.txt
105
63
  - README.md
106
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
107
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
108
96
  - lib/torchvision/transforms/compose.rb
109
97
  - lib/torchvision/transforms/functional.rb
110
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
111
103
  - lib/torchvision/transforms/to_tensor.rb
104
+ - lib/torchvision/utils.rb
112
105
  - lib/torchvision/version.rb
113
106
  homepage: https://github.com/ankane/torchvision
114
107
  licenses:
115
108
  - BSD-3-Clause
116
109
  metadata: {}
117
- post_install_message:
110
+ post_install_message:
118
111
  rdoc_options: []
119
112
  require_paths:
120
113
  - lib
@@ -122,15 +115,15 @@ required_ruby_version: !ruby/object:Gem::Requirement
122
115
  requirements:
123
116
  - - ">="
124
117
  - !ruby/object:Gem::Version
125
- version: '2.4'
118
+ version: '2.6'
126
119
  required_rubygems_version: !ruby/object:Gem::Requirement
127
120
  requirements:
128
121
  - - ">="
129
122
  - !ruby/object:Gem::Version
130
123
  version: '0'
131
124
  requirements: []
132
- rubygems_version: 3.1.2
133
- signing_key:
125
+ rubygems_version: 3.2.3
126
+ signing_key:
134
127
  specification_version: 4
135
128
  summary: Computer vision datasets, transforms, and models for Ruby
136
129
  test_files: []