mmsegmentation/mmseg/models/segmentors
Ziyi Wu 768f704517
[Enhance] Replace data_dict calling 'img' key to support MMDet3D (#514)
* remove dict calling img key for compatibility

* fix unit test

* infer batch size using len(result) to be consistent with mmcv
2021-04-29 01:01:34 -07:00
..
__init__.py add BaseSegmentor import to segmentors/__init__.py (#495) 2021-04-19 08:52:42 -07:00
base.py [Enhance] Replace data_dict calling 'img' key to support MMDet3D (#514) 2021-04-29 01:01:34 -07:00
cascade_encoder_decoder.py init commit 2020-07-10 02:39:01 +08:00
encoder_decoder.py add dynamic export and visualize to pytorch2onnx (#463) 2021-04-12 11:54:59 -07:00