- 21 Oct, 2019 2 commits
-
-
F-G Fernandez authored
* test: Updated asserts in test_io Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483 * test: Fixed test_io In order to use unittest.TestCase for asserts in temp_video, the object has to be passed itself to temp_video. * test: Updated asserts to AssertionError Switched initial raw assert to AssertionError. * test: Changed AssertionError for ValueError
-
F-G Fernandez authored
Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
-
- 18 Oct, 2019 7 commits
-
-
ekka authored
* Make crop torchscriptable Relevant #1375 * Invert x and y axis * fix lint * Add crop test * revert deletion of space in functional * add import random * add dimension in doc * add import * fix flake8 * change to self.assert* * convert to uint8 * assertTrue * lint
-
F-G Fernandez authored
Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
-
F-G Fernandez authored
Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
-
Lara Haidar authored
* onnx esport faster rcnn * test * address PR comments * revert unbind workaround * disable tests for older versions of pytorch
-
F-G Fernandez authored
Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483
-
Francisco Massa authored
This reverts commit 1e857d93.
-
F-G Fernandez authored
* test: Updated assert in test_transforms Updated all raw asserts to corresponding unittest.TestCase.assert. See #1483 * test: Fixed linter on test_transforms
-
- 17 Oct, 2019 1 commit
-
-
fbbradheintz authored
* added correctness tests for classification models * refactored tests for extensibility & usability * flake8 fixes
-
- 16 Oct, 2019 2 commits
-
-
Surgan Jandial authored
* vflip and hflip tensor * vflip and hflip tensor * changes made * lint * lint * lint failing
-
Lukas Bommes authored
* added PSRoiAlign and PSRoiPool with C++ autograd and torch ops * fixed linter errors * fixed linter errors 2 * fixed linter errors 3
-
- 15 Oct, 2019 2 commits
-
-
Francisco Massa authored
* Handle corrupted video headers in io * Catch exceptions while decoding partly-corrupted files * Add more tests
-
Lara Haidar authored
* Support Exporting RPN to ONNX * address PR comments * fix cat * add flatten * replace cat by stack * update test to run only on rpn module * use tolerate_small_mismatch
-
- 12 Oct, 2019 1 commit
-
-
Zhicheng Yan authored
* extend video reader to support fast video probing * fix c++ lint * small fix * allow to accept input video of type torch.Tensor
-
- 08 Oct, 2019 1 commit
-
-
Francisco Massa authored
-
- 04 Oct, 2019 2 commits
-
-
Lara Haidar authored
* Support Exporting MultiScaleRoiAlign to ONNX * remove cast * fix dtype * move cast
-
Zhicheng Yan authored
* move sampler into TV core. Update UniformClipSampler * Fix reference training script * Skip test if pyav not available * change interpolation from round() to floor() as round(0.5) behaves differently between py2 and py3
-
- 03 Oct, 2019 1 commit
-
-
Zhicheng Yan authored
* add metadata to video dataset classes. bug fix. more robustness * query video backend within VideoClips class * Fix tests * Fix lint
-
- 02 Oct, 2019 1 commit
-
-
eellison authored
* add support for video resnet models, restructure script test to just ignore RCNN models * switch back to testing subset of the models
-
- 01 Oct, 2019 1 commit
-
-
eellison authored
* add expected result tests * fix wrong assertion * start with only detection models * remove unneeded rng setting * fix test * add tuple support * update test * syntax error * treat .pkl files as binary data, see : https://git-scm.com/book/en/v2/Customizing-Git-Git-Attributes#_binary_files * fix test * fix elif * Map tensor results and enforce maximum pickle size * unrelated change * larger rtol * pass rtol atol around * last commit i swear... * respond to comments * fix flake * fix py2 flake
-
- 30 Sep, 2019 1 commit
-
-
Chandresh Kanani authored
modified code of io.read_video and io.read_video_timestamps to intepret pts values in seconds (#1331) * modified code of io.read_video and io.read_video_timestamps to interpret pts values in seconds * changed default value for pts_unit to pts, corrected formatting * hanndliing both fractions and floats for start_pts and end_pts, added test cases for pts_unit sec * moved unit conversion logic to _read_from_stream method
-
- 27 Sep, 2019 1 commit
-
-
eellison authored
* make googlnet scriptable * Remove typing import in favor of torch.jit.annotations * add inceptionnet * flake fixes * fix asssert true * add import division for torchscript * fix script compilation * fix flake, py2 division error * fix py2 division error
-
- 26 Sep, 2019 1 commit
-
-
Francisco Massa authored
* Fix nightly builds * Fix lint * Disable video_reader tests on Travis * Disable one more test that segfaults on Travis * Fix typo in expression
-
- 24 Sep, 2019 4 commits
-
-
Zhicheng Yan authored
* add _backend argument to __init__() of class VideoClips * minor fix * minor fix * Make backend private in VideoClips * Fix lint
-
Zhicheng Yan authored
* video transforms * [video transforms]in ToTensorVideo, divide value by 255.0 * [video transforms] fix a bug * fix linting * Make changes backwards-compatible
-
Ardalan authored
* add tar.xz archive handler * update unittest for tar.xz archive * remove .tar.xz unittest * add separate .tar.xz unittest * update PY2 compatibility
-
Francisco Massa authored
-
- 23 Sep, 2019 1 commit
-
-
Surgan Jandial authored
* format param added * lint fixes * lint fixes * lint fixes
-
- 20 Sep, 2019 3 commits
-
-
Zhicheng Yan authored
* [video reader] inception commit * add method save_metadata to class VideoClips in video_utils.py * add load_metadata() method to VideoClips class * add Exception to not catch unexpected events such as memory erros, interrupt * fix bugs in video_plus.py * [video reader]remove logging. update setup.py * remove time measurement in test_video_reader.py * Remove glog and try making ffmpeg finding more robust * Add ffmpeg to conda build * Add ffmpeg to conda build [again] * Make library path finding more robust * Missing import * One more missing fix for import * Py2 compatibility and change package to av to avoid version conflict with ffmpeg * Fix for python2 * [video reader] support to decode one stream only (e.g. video/audio stream) * remove argument _precomputed_metadata_filepath * remove save_metadata method * add get_metadata method * expose _precomputed_metadata and frame_rate arguments in video dataset __init__ method * remove ssize_t * remove size_t to pass CI check on Windows * add PyInit__video_reader function to pass CI check on Windows * minor fix to define PyInit_video_reader symbol * Make c++ video reader optional * Temporarily revert changes to test_io * Revert changes to python files * Rename files to make it private * Fix python lint * Fix C++ lint * add a functor object EnumClassHash to make Enum class instances usable as key type of std::unordered_map * fix cpp format check
-
eellison authored
* script_fcn_resnet * Make old models load * DeepLabV3 also got torchscript-ready
-
eellison authored
* make densenet scriptable * make py2 compat * use torch List polyfill * fix unpacking for checkpointing * fewer changes to _Denseblock * improve error message * print traceback * add typing dependency * add typing dependency to travis too * Make loading old checkpoints work
-
- 18 Sep, 2019 3 commits
-
-
Francisco Massa authored
* Remove C++ extensions in favor of custom ops * Remove unused custom_ops.cpp file * Rename _custom_ops.py * Reorganize functions * Minor improvements and fixes * Fix lint * Fully scriptable ops * Import types used by annotations
-
Francisco Massa authored
* Add Hub tests in torchvision * Run test_hub in its own interpreter
-
Francisco Massa authored
* Make AnchorGenerator support half precision * Add test for fasterrcnn with double * convert gt_boxes to right dtype
-
- 17 Sep, 2019 2 commits
-
-
Lara Haidar authored
* Support Exporting GeneralizedRCNNTransform * refactor code to address comments * update tests * address comments * revert min_size to test CI * re-revert min_size
-
eellison authored
-
- 10 Sep, 2019 2 commits
-
-
Thomas Viehmann authored
* Make custom ops differentiable and replace autograd.Function. Use ops unconditionally. We may consider removing the extension functions in a follow-up. The code-path is tested by the exisitng tests for differentiability. * add scripting gradchecks tests and use intlist * fix implicit tuple conversion for gcc-5 * fix merge
-
Lara Haidar authored
* Revert "Revert "Register Torchvision Ops as Cutom Ops (#1267)" (#1316)" This reverts commit fe234fc8. * Make import of C++ extensions lazy * define python initialization functions for extension * Fix lint
-
- 09 Sep, 2019 1 commit
-
-
Edward Z. Yang authored
This reverts commit 78f169b5.
-