Qiaofei Li
06556c8459
Add shear augmentation ( #526 )
...
* Add shear augmentation
* Update geometric.py
* Update geometric.py
2020-09-02 10:57:36 +08:00
Wang Xinjiang
15b37b0b78
Allow imshow_det_bboxes to return image with bboxes. ( #527 )
2020-09-02 00:39:57 +08:00
Wenwei Zhang
fc4993cdec
[feature] Use cu92 & ubuntu1604 for torch 1.3.1 ( #524 )
...
* [feature] Use cu92 & ubuntu1604 for torch 1.3.1
* [fix]Use ubuntu-16.04 rather than latest for cu92
* [fix]: soft link cuda path
* Debug cu101 bug
* Update usr/include for cublas with cu101
* Check default nvcc version
* rm :
* add cuda root
* Try to locate cublas_v2.h
* Set cublas version
* Force cublas version
* check usr/local/cuda-10.2
* Add tree
* cp cublas files
* cp cublas files to lib64
* do not tree cuda
* fix path cp error
* cp dir
* Keep using CUDA10.1 only
* recover empty line
2020-09-01 16:15:03 +08:00
Xiaojie Li
c3d8eb34ff
add Swish activation ( #522 )
...
* update impad
* fix docstring
* add shape for impad
* fix unit test
* remove old version & fix doc
* fix linting
* fix doc
* add linear decay learning rate scheduler
* fix impad
* fix setup.cfg
* fix linting
* add yapf
* add swish
* fix lr_updater
* fix lr_updater.py
* update swish
* add swish
* fix inplace
* fix typo
Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-27 00:39:17 +08:00
Kai Chen
66a38c86f8
Move unit tests to specific folders ( #520 )
...
* move unit tests to specific folders
* fix path error
* remove some assertions
* fix ignore path
2020-08-25 20:11:56 +08:00
Kai Chen
89e1716afe
Bump version to v1.1.1 ( #518 )
...
* bump version to 1.1.1
* add more version utils
* move parse_version_info to version.py
2020-08-24 16:30:45 +08:00
Wang Xinjiang
09b7d6c7dd
Import modules from a string list ( #514 )
...
* Custom imports
* Resolve comments
* Add unittest
* Add unittest
* Rename custom_imports to import_modules_from_strings
* Move import_modules_from_strings ito misc.py and allow failed imports
* small change
* small change
* change mmcv.runner to os.path
2020-08-24 13:45:34 +08:00
Kai Chen
270e470ee1
[feature] Add collect_env() to collect environment info ( #517 )
...
* [feature] Add collect_env() to collect environment info
* fix unit tests
* fix the case when ops are not compiled
* fix docstring
2020-08-24 13:41:54 +08:00
Matthew Dawkins
aebdcb6661
Fix inconsistent return types in pybind11 function prototypes ( #509 )
...
* Fix function types
* Remove const
2020-08-24 12:46:04 +08:00
Wenwei Zhang
f4a5446e15
Support to split batched_nms when box number is too large ( #516 )
...
* Support to split batched_nms when box number is too large
* mv data from gpu to cpu
* Set split_thr through nms_cfg
* clean code
* Update motivation in docstring
* fix typos
2020-08-24 00:32:39 +08:00
Jerry Jiarui XU
83d9a9c89b
[Feature] Add diagonal flip ( #515 )
2020-08-22 14:29:27 +08:00
Matthew Dawkins
eb0414f495
Fix windows compile issues ( #510 )
2020-08-22 14:24:16 +08:00
John Zhu
5e7f785364
fix regex bugs in Widows paths ( #513 )
2020-08-21 14:04:42 +08:00
Kevin
95a9728c42
Support pickle.loads ( #499 ) ( #500 )
2020-08-21 13:06:00 +08:00
Kai Chen
77c03f44b8
bump version to 1.1.0 ( #508 )
2020-08-20 01:05:28 +08:00
robin Han
11d8dd533d
support ONNX adaptive average pooling ( #504 )
...
* support ONNX adaptive average pooling
* fix double quotes
Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-08-20 00:43:35 +08:00
ychan
5e3f56f8a6
fix mdconv addmm bug for parrots ( #450 )
...
* fix mdconv addmm bug for parrots
* fix mdconv ctv save tensor
Co-authored-by: hanyachao <hanyachao@sensetime.com>
2020-08-19 23:58:14 +08:00
Jerry Jiarui XU
6159dac24c
Update darknet url ( #507 )
2020-08-19 18:29:49 +08:00
su
eb65c21da2
Fix issue#505 ( #506 )
2020-08-19 16:13:20 +08:00
Wenwei Zhang
7a6285b190
Fix mmcv-dataparallel ( #497 )
...
* Fix mmcv-dataparallel
* Fix (parallel): fix CPU inference with MMDataParallel
* Update docstrings
* Doc (parallel): refine docstrings
* Fix (parallel): fix missing changes of train/val step function
* resolve comments
* Fix (data_parallel): fix bug when single gpu test return None
2020-08-19 14:22:40 +08:00
Jintao Lin
4ec73abbcc
Logging mode by using runner.mode ( #495 )
...
* log mode by using runner.mode
* add IterTimerHook to fix unittest
* correct the logic
* display fix when using EvalHook in train mode
* simplify logic
2020-08-17 18:50:19 +08:00
Cao Yuhang
7b18b97791
fix saconv ( #489 )
...
* fix saconv
* add parrots condition
* add unittest
* fix torch version
2020-08-16 01:45:01 +08:00
Wang Xinjiang
eacaf475f7
fix some pavi logger hooks ( #481 )
...
* fix some pavi logger hooks
* fix unittest
* fix small bugs
* small change
* fix unittest
* Add EpochBasedRunner conditions
* Add session text
* fix small bug
* fetch runner mode from log buffer
* Add max_iter to pavi session text
* change yaml.dump to yamp.dump(yaml.load(mmcv.dump))
* Directly use by_epoch
* fix unittest
* add comments
* Use runner.epoch + 1 in pavi log
* fix runner.epoch issue for runner.mode=='val'
* fix runner.epoch issue for runner.mode=='val'
* Use abspath instead of realpath
* Add meta dump unittest
* small change
* Add comments
2020-08-16 01:20:08 +08:00
Jintao Lin
c8e85b28e4
Add tin_shift function ( #492 )
...
* add tin shift
* add unittest
* add docstring
* add docstring
* parrots for tin_shift
* fix lint
* fix lint
Co-authored-by: jiaomenglei <jiaomenglei@sensetime.com>
2020-08-15 22:29:07 +08:00
Jerry Jiarui XU
15537c5a39
Add LOCAL_RANK env var for slurm ( #496 )
2020-08-14 20:35:05 +08:00
Wang Xinjiang
d19fad02ba
Add json set_default for unserializable values ( #485 )
...
* Add json set_default for unserializable values
* kwargs.setdefault
2020-08-13 22:05:43 +08:00
Cao Yuhang
dc778481cb
add op trouble shooting ( #479 )
...
* add op trouble shooting
* update trouble_shooting.md
* clean ops.md
* add trouble shooting to index.rst
* reorder
* add troubleshooting in readme
2020-08-13 22:04:58 +08:00
Wenwei Zhang
5ade35f4cf
use elif rather than if ( #491 )
2020-08-13 20:38:27 +08:00
Jerry Jiarui XU
4cc48073bc
Change pretrain url from aws to aliyun ( #490 )
2020-08-13 19:32:28 +08:00
Zijian He
e7e0c89f5c
fix the bug ( #472 ) ( #484 )
...
* fix the bug (#472 )
* fix the bug (#472 )
* fix the bug (#472 )
Co-authored-by: hezijian <hezijian@dm-ai.cn>
2020-08-13 19:30:28 +08:00
Yuanhao Zhu
51c65c97ec
fix syncbn parameter order mismatch and parrots bug ( #488 )
2020-08-13 19:27:05 +08:00
su
17e4732c49
Change the epoch runner to use the data_loader from attributes rather than args. ( #483 )
2020-08-12 22:09:07 +08:00
Kai Chen
e92f826abc
close the config file after opening ( #480 )
2020-08-11 21:28:44 +08:00
Kai Chen
cac22f8cf5
Add pytorch 1.6 to CI ( #476 )
...
* add pytorch 1.6 to CI
* fix typo
* fix ci error
Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
2020-08-09 21:56:15 +08:00
Jerry Jiarui XU
56e71a718b
Add Depthwise Seperable ConvModule ( #477 )
2020-08-09 17:29:02 +08:00
Kai Chen
530ae20046
add pre-build packages for pytorch 1.6 ( #475 )
2020-08-09 15:44:44 +08:00
Jerry Jiarui XU
9b36c1ab3e
Fixed iter_base_runner logger ( #474 )
2020-08-09 00:04:48 +08:00
Jintao Lin
edbbc0f215
fix bug for self.warmup_iters when warmup_by_epoch ( #470 )
2020-08-06 22:07:31 +08:00
ChaseMonsterAway
6b43ae3700
Update cnn.md ( #466 )
2020-08-06 16:51:58 +08:00
Danil
0a1cea73c1
Add dot to yml extension ( #469 )
2020-08-06 16:51:32 +08:00
Kai Chen
045629c280
Fix torch installation in github action ( #462 )
...
* bump version to 1.0.5
* fix pytorch name
* fix version
* disable the cache
2020-08-03 23:10:39 +08:00
Wenwei Zhang
d36850f517
Add git hash ( #461 )
...
* Add git hash
* Add fallback
* Add func in mmcv.utils
2020-08-03 21:00:33 +08:00
Kai Chen
60ce48d164
bump version to 1.0.5 ( #459 )
2020-08-03 13:17:03 +08:00
Jintao Lin
71b59a8da0
assign self.warmup_iters in before_run function ( #458 )
2020-08-03 11:55:51 +08:00
Jintao Lin
9de04d2284
Replace re with regex ( #457 )
...
* replace `re` with `regex`
* add regex in requirements
* use regex on windows and re on linux
* update requirements.txt
2020-08-03 11:45:50 +08:00
Jerry Jiarui XU
66864d4abc
Update pretrain link ( #455 )
...
* update pretrain link
* remove some
2020-08-02 22:00:05 +08:00
shilong
1830347f8b
Ema ( #421 )
...
* add ema hook
* add ema hook resume
* add ema hook test
* fix typo
* fix according to comment
* delete logger
* fix according to comment
* fix unitest
* fix typo
* fix according to comment
* change to resume_from
* typo
* fix isort
2020-07-30 22:06:19 +08:00
Cao Yuhang
4676031c8b
Move onnx op import inside symbolic ( #453 )
...
* add pat
* move onnx op import to symbolic
2020-07-28 23:14:36 +08:00
Jintao Lin
829d1e381d
add function and class related with fp16 to runner/__init__.py ( #448 )
2020-07-27 10:43:20 +08:00
Wang Xinjiang
d4da3daa7e
Syncbuf ( #447 )
...
* More robust sync buffer hook
* More robust sync buffer hook
* Reformat
2020-07-25 12:51:46 +08:00
