mmengine

Commit Graph

Author	SHA1	Message	Date
Mashiro	d624fa9191	[Enhance] assert image shape before forward (#300 ) * assert image shape before forward * add unit test * enhance error message * allow gray image input * fix as comment * fix unit test * fix unit test	2022-06-28 11:46:12 +08:00
RangiLyu	819e10c24c	[Fix] Fix image dtype when enable_normalize=False. (#301 ) * [Fix] Fix image dtype when enable_normalize=False. * update ut * move to collate * update ut	2022-06-13 21:21:19 +08:00
Mashiro	931db99005	[Enhance] Enhance img data preprocessor (#290 ) * fix BaseDataPreprocessor * fix BaseDataPreprocessor * change device type to torch.device * change device type to torch.device * fix cpu method of base model * Allow ImgDataPreprocessor do not normalize * remove unnecessary type ignore * make mean and std optional * refine docstring	2022-06-09 20:12:15 +08:00
Mashiro	a9afdad7a8	[Fix] Fix BaseDataPreprocessor and BaseModel (#285 ) * fix BaseDataPreprocessor * fix BaseDataPreprocessor * change device type to torch.device * change device type to torch.device * fix cpu method of base model	2022-06-09 11:45:19 +08:00
Mashiro	6ee675430f	[Refactor]: change order of BaseModel arguments (#282 )	2022-06-08 13:28:00 +08:00
Mashiro	f04fec736d	[Feature]: add base model, ddp model wrapper and unit test (#268 ) * add base model, ddp model and unit test * add unit test * fix unit test * fix docstring * fix cpu unit test * refine base data preprocessor * refine base data preprocessor * refine interface of ddp module * remove optimizer hook * add forward * fix as comment * fix unit test * fix as comment * fix build optimizer wrapper * rebase main and fix unit test * stack_batch support stacking ndim tensor, add docstring for merge dict * fix lint * fix test loop * make precision_context effective to data_preprocessor * fix as comment * fix as comment * refine docstring * change collate_data output typehints * rename to_rgb to bgr_to_rgb and rgb_to_bgr * support build basemodel with built DataPreprocessor * fix as comment * fix docstring	2022-06-07 22:13:53 +08:00

6 Commits (d624fa9191e0f1a5c79f366a03d809a7a6be3859)