- 11 May, 2020 1 commit
-
-
Philip Meier authored
* add mypy config * fix syntax error * fix annotations in torchvision/utils.py * add mypy type check to CircleCI * add mypy cache to ignore files * try fix CI * ignore flake8 F821 since it interferes with mypy * add mypy type check to config generator * explicitly set config files
-
- 07 May, 2020 3 commits
-
-
Francisco Massa authored
* Fix mypy type annotations * follow torchscript Tuple type * redefine torch_choice output type * change the type in cached_grid_anchors * minor bug Co-authored-by:
Guanheng Zhang <zhangguanheng@devfair0197.h2.fair> Co-authored-by:
Guanheng Zhang <zhangguanheng@learnfair0341.h2.fair>
-
Negin Raoof authored
* Fixing nms on boxes when no detection * test * Fix for scale_factor computation * remove newline * Fix for mask_rcnn dynanmic axes * Clean up * Update transform.py * Fix for torchscript * Fix scripting errors * Fix annotation * Fix lint * Fix annotation * Fix for interpolate scripting * Fix for scripting * refactoring * refactor the code * Fix annotation * Fixed annotations * Added test for resize * lint * format * bump ORT * ort-nightly version * Going to ort 1.1.0 * remove version * install typing-extension * Export model for images with no detection * Upgrade ort nightly * update ORT * Update test_onnx.py * updated tests * Updated tests * merge * Update transforms.py * Update cityscapes.py * Update celeba.py * Update caltech.py * Update pkg_helpers.bash * Clean up * Clean up for dynamic split * Remove extra casts * flake8
-
Guillem Orellana Trullols authored
Now the dataset is not working properly because of this line of code `indices = [i for i in range(len(video_list)) if video_list[i][len(self.root) + 1:] in selected_files]`. Performing the `len(self.root) + 1` only make sense if there is no training / to root ``` >>> root = 'data/ucf-101/videos' >>> video_path = 'data/ucf-101/videos/activity/video.avi' >>> video_path [len(root ):] '/activity/video.avi' >>> video_path [len(root ) + 1:] 'activity/video.avi' ``` Appending the root path also to the selected files is a simple solution and make the dataset works with and without a trailing slash.
-
- 05 May, 2020 5 commits
-
-
Ashish Malhotra authored
-
Bisakh Mondal authored
-
F-G Fernandez authored
* feat: Added number of features in FrozenBatchNorm2d repr While BatchNorm layers have extensive information in their repr, FrozenBatchNorm2d has one * refactor: Refactored FrozenBatchNorm2d __repr__ * test: Added unittest for FrozenBatchNorm2d __repr__ * style: Removed blank lines in test_ops * refactor: Avoids creating an extra attribute for __repr__ * style: Switched __repr__ to f-string Since support of Python version ealier than 3.6 have been dropped, f-string can be used. * fix: Fixed typo in __repr__ * style: Switched unittest .format to f-string
-
Francisco Massa authored
-
Hong Xu authored
`Mn`, `Sn` are used as mean and std, but their suddenly turned to be `mean[n]` and `std[n]` in about 10 words later
-
- 04 May, 2020 5 commits
-
-
Arash Javanmard authored
At the moment in the ASPP-Layer the number of output channels are predefined as a constant, which is good for DeepLab but not necessairly in other projects, where another out-channel Nr. is required. Also the number of "atrous rates" is fixed to three, which also could be sometimes more or less depending on the notwork-arch. Again these fixed values may make sense in DeepLab-Model but not necessarily in other type of models. This pull-req. contains the needed changes to make ASPP-Layer generic.
-
Oscar Mañas authored
-
Peter Steinbach authored
-
Gao, Xiang authored
* Don't include CUDAApplyUtils.cuh * fix format * fix atomic
-
Fahri Ali Rahman authored
* Improve documentation for NMS * update nms doc for special case
-
- 29 Apr, 2020 1 commit
-
-
theonekeyg authored
-
- 28 Apr, 2020 1 commit
-
-
Jackson Liu authored
-
- 27 Apr, 2020 1 commit
-
-
Philip Meier authored
* only display extensions if available * add tests * fix lint
-
- 26 Apr, 2020 1 commit
-
-
Francisco Massa authored
-
- 24 Apr, 2020 1 commit
-
-
Potter Hsu authored
* For detection model, replace L1 loss with beta smooth L1 loss to achieve better performance. * Add type annotations for torchscript * Resolve E226
-
- 23 Apr, 2020 1 commit
-
-
Yuxin Wu authored
* fix the use of contiguous() in kernels * clang-format * add a contiguous in nms Co-authored-by:Yuxin Wu <ppwwyyxx@users.noreply.github.com>
-
- 21 Apr, 2020 1 commit
-
-
Negin Raoof authored
* Fixing nms on boxes when no detection * test * Fix for scale_factor computation * remove newline * Fix for mask_rcnn dynanmic axes * Clean up * Update transform.py * Fix for torchscript * Fix scripting errors * Fix annotation * Fix lint * Fix annotation * Fix for interpolate scripting * Fix for scripting * refactoring * refactor the code * Fix annotation * Fixed annotations * Added test for resize * lint * format * bump ORT * ort-nightly version * Going to ort 1.1.0 * remove version * install typing-extension
-
- 16 Apr, 2020 1 commit
-
-
dwSun authored
-
- 14 Apr, 2020 1 commit
-
-
theotheo authored
-
- 10 Apr, 2020 1 commit
-
-
moto authored
-
- 09 Apr, 2020 2 commits
-
-
Francisco Massa authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/2077 Pull Request resolved: https://github.com/facebookresearch/SlowFast/pull/164 This is a follow-up diff from D18720474 We will be releasing a new version of torchvision soon and the signature of those functions is not ready yet, following my comment in https://our.intern.facebook.com/intern/diff/D18720474/?transaction_id=561239541337402 Reviewed By: stephenyan1231 Differential Revision: D20914571 fbshipit-source-id: 1a7560b8f8e46ab42ef376c50b494a4f73923e94 Co-authored-by:
Francisco Massa <fmassa@fb.com>
-
Luan Pham authored
-
- 07 Apr, 2020 4 commits
-
-
Brian Hart authored
Torchvision includes at least 3 bits of code that calculate box Intersection over Union values (and usually compare to a threshold): - box_iou in torchvision/ops/boxes.py - devIoU in torchvision/csrc/cuda/nms_cuda.cu - nms_cpu_kernel in torchvision/csrc/cpu/nms_cpu.cpp The calculations were performed slightly differently between those, leading to occasional differences in results. Update devIoU to use the same method as the others for better consistency. This change improves agreement between the CPU and CUDA calculations but the results can still differ slightly. Setting NVCC_FLAGS to include "--fmad=true" would provide still better agreement, but with likely cost to performance.
-
Philip Meier authored
-
Francisco Massa authored
* Add tests for negative samples for Mask R-CNN and Keypoint R-CNN * Fix lint
-
AhnDW authored
* Replace **.is_cuda() to just is_cuda() * Replace type to scalar_type * Fix lint, clang-format * Fix lint, clang-format
-
- 06 Apr, 2020 2 commits
-
-
Yonghye Kwon authored
cleanup
-
Philip Meier authored
* type annotations for torchvision/utils.py * add missing annotation for make_grid * fix annotation for save_image * mirror PIL annotation for fp
-
- 03 Apr, 2020 4 commits
-
-
Francisco Massa authored
-
gslotman authored
-
Francisco Massa authored
* Add clang-format to CircleCI * Fix for clang-format version * Fix lint and remove Travis CI * Seeing if lost commit comes back * Fix lint * Re-enable all tests
-
Francisco Massa authored
* Add CircleCI job for python lint * Break lint * Fix * Fix lint * Re-enable all tests and remove travis python lint
-
- 02 Apr, 2020 2 commits
-
-
Francisco Massa authored
Summary: Fix docstring formatting issues Reviewed By: fmassa Differential Revision: D20736644 fbshipit-source-id: 78f66045cfd4c84cb35ca84a1e1fa6aadcd50642 Co-authored-by:Patrick Labatut <plabatut@fb.com>
-
Francisco Massa authored
* Add test for large batches in DeformConv2d * Clean-up and (try) fix DeformConv2d * Simplifications and bugfixes * Try fix CUDA now
-
- 01 Apr, 2020 2 commits
-
-
Francisco Massa authored
-
Philip Meier authored
-