torchvision 0.1.1 → 0.2.2

Sign up to get free protection for your applications and to get access to all the features.
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: []