- 03 Jun, 2019 2 commits
-
-
Philip Meier authored
* added fake data * fixed fake data * renamed extract and download methods and added functionality * added raw fake data * refactored imagenet and added test * flake8 * added fake devkit and mocked download_url * reversed uncommenting * added mock to CI * fixed tests for imagefolder * flake8
-
Gu-ni-kim authored
Add 'import torch' in example
-
- 31 May, 2019 1 commit
-
-
Ailing authored
* add more hub models * fix lint
-
- 30 May, 2019 1 commit
-
-
Soumith Chintala authored
* make C extension lazy-import * add lazy loading to roi_pool
-
- 29 May, 2019 3 commits
-
-
Edouard Oyallon authored
* modif of the STL10 loader * missing space
-
Francisco Massa authored
* Fix STL10 repr * Do not inherit from Cifar10 * Make it safer to inherit from VisionDataset
-
Francisco Massa authored
* WIP * WIP: minor improvements * Add tests * Fix typo * Use download_and_extract on caltech, cifar and omniglot * Add a print message during extraction * Remove EMNIST from test
-
- 27 May, 2019 2 commits
-
-
Tongzhou Wang authored
-
Francisco Massa authored
* add USPS dataset * minor fixes * Improvements to the USPS dataset Add it to the documentation, expose it to torchvision.datasets and inherit from VisionDataset
-
- 25 May, 2019 1 commit
-
-
7d authored
Consider the difference of the division operator between Python 2.x and Python 3.x.
-
- 24 May, 2019 1 commit
-
-
Konstantin Lopuhin authored
* Allow forcing GPU build with FORCE_CUDA=1 This is convenient to e.g. build with GPU support inside a docker image * Document FORCE_CUDA
-
- 23 May, 2019 3 commits
-
-
Francisco Massa authored
* #944 MSBuild Compile time casting Error * #944 MSBuild Error static_cast<Long> to static_cast<int64_t> * Add eval.py Not Work find_contours * Remove unnecessary file * Lint
-
peterjc123 authored
-
Varun Agrawal authored
Updated nms_cuda signature to accept detections and scores as separate tensors. This also required updating the indexing in the NMS CUDA kernel. Also made the iou_threshold parameter name consistent across implementations.
-
- 22 May, 2019 8 commits
-
-
Soumith Chintala authored
-
Soumith Chintala authored
This reverts commit 06b9b0fc.
-
Francisco Massa authored
-
Sergey Zagoruyko authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
- 21 May, 2019 8 commits
-
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
Allows for easily evaluating the pre-trained models in the modelzoo
-
Francisco Massa authored
This makes it consistent with the other models, which returns nouns in plurial
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
Also adds documentation for the segmentation models
-
- 20 May, 2019 5 commits
-
-
Francisco Massa authored
* Add more documentation for the ops * Add documentation for Faster R-CNN * Add documentation for Mask R-CNN and Keypoint R-CNN * Improve doc for RPN * Add basic doc for GeneralizedRCNNTransform * Lint fixes
-
Francisco Massa authored
Those were not free parameters, and can be inferred via the size of the output feature map
-
Francisco Massa authored
* Add COCO pre-trained weights for Faster R-CNN R-50 FPN * Add weights for Mask R-CNN and Keypoint R-CNN
-
Francisco Massa authored
This reverts commit b384c4e7.
-
Francisco Massa authored
-
- 19 May, 2019 5 commits
-
-
Francisco Massa authored
* Split mask_rcnn.py into several files * Lint
-
Francisco Massa authored
-
Francisco Massa authored
-
Francisco Massa authored
* Move segmentation models to its own folder * Add missing files
-
ekka authored
* Remove dependency from functool in ShuffleNetsV2 This PR removes the dependence of the ShuffleNetV2 code from `functool`. * flake fix
-