Skip to content

Commit 6d158ad

Browse files
authored
Update README.md
1 parent 470220b commit 6d158ad

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed

README.md

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,3 +174,35 @@ See the [documentation](https://rwightman.github.io/pytorch-image-models/)
174174
## Train, Validation, Inference Scripts
175175

176176
The root folder of the repository contains reference train, validation, and inference scripts that work with the included models and other features of this repository. They are adaptable for other datasets and use cases with a little hacking. See [documentation](https://rwightman.github.io/pytorch-image-models/scripts/) for some basics and [training hparams](https://rwightman.github.io/pytorch-image-models/training_hparam_examples) for some train examples that produce SOTA ImageNet results.
177+
178+
## Awesome PyTorch Resources
179+
180+
One of the greatest assets of PyTorch is the community and their contributions. A few of my favourite resources that pair well with the models and componenets here are listed below.
181+
182+
### Training / Frameworks
183+
* PyTorch Lightning - https://github.com/PyTorchLightning/pytorch-lightning
184+
* fastai - https://github.com/fastai/fastai
185+
186+
### Computer Vision / Image Augmentation
187+
* Albumentations - https://github.com/albumentations-team/albumentations
188+
* Kornia - https://github.com/kornia/kornia
189+
190+
### Metric Learning
191+
* PyTorch Metric Learning - https://github.com/KevinMusgrave/pytorch-metric-learning
192+
193+
### Object Detection, Instance and Semantic Segmentation
194+
* Detectron2 - https://github.com/facebookresearch/detectron2
195+
* Segmentation Models (Semantic) - https://github.com/qubvel/segmentation_models.pytorch/issues
196+
* EfficientDet (Obj Det, Semantic soon) - https://github.com/rwightman/efficientdet-pytorch
197+
198+
## Licenses
199+
200+
### Code
201+
The code here is licensed Apache 2.0. I've taken care to make sure any third party code included or adapted has compatible (permissive) licenses such as MIT, BSD, etc. I've made an effort to avoid any GPL / LGPL conflicts. That said, it is your responsibility to ensure you comply with license here and conditions of any dependent licenses. Where applicable, I've linked the sources/references for various components in docstrings. If you think I've missed anything please create an issue.
202+
203+
### Pretrained Weights
204+
So far all of the pretrained weights available here are pretrained on ImageNet with a select few that have some additional pretraining (see extra note below). ImageNet was released for non-commercial research purposes only (http://www.image-net.org/download-faq). It's not clear what the implications of that are for the use of pretrained weights from that dataset. Any models I have trained with ImageNet are done for research purposes and one should assume that the original dataset license applies to the weights. It's best to seek legal advice if you intend to use the pretrained weights in a commercial product.
205+
206+
#### Pretrained on more than ImageNet
207+
Several weights included or references here were pretrained with proprietary datasets that I do not have access to. These include the Facebook WSL, SSL, SWSL ResNe(Xt) and the Google Noisy Student EfficientNet models. The Facebook models have an explicit non-commercial license (CC-BY-NC 4.0, https://github.com/facebookresearch/semi-supervised-ImageNet1K-models, https://github.com/facebookresearch/WSL-Images). The Google models do not appear to have any restriction beyond the Apache 2.0 license (and ImageNet concerns). In either case, you should contact Facebook or Google with any questions.
208+

0 commit comments

Comments
 (0)