Commit Graph

11 Commits (5b02db5b2b83f1f7a4bc66ae642da201b9a65be2)

Author SHA1 Message Date
Kapulkin Stanislav 5b02db5b2b [+] save model state dict and load it 2022-02-09 17:53:40 +03:00
Kapulkin Stanislav 4e2f7ddfbe [*] image resizing in dataset 2022-02-08 19:24:30 +03:00
Kapulkin Stanislav b6f9f63d83 [+] train loop scripts 2022-02-07 00:19:06 +03:00
Zhanghan Ke 3850cf6106
update SOC code 2021-05-18 03:01:51 +08:00
kevin 5da2bb6c81 support TorchScript 2021-03-12 17:10:47 +08:00
ZHKKKe 44667be551 add ONNX version of MODNet 2021-02-19 14:48:22 +08:00
manthandrive01 bffa00eebd Inference with onnx
export model and run inference
2021-02-11 23:52:28 +05:30
Zhanghan Ke c432f592e7
update the comments 2021-01-28 18:40:25 +08:00
ZHKKKe 9106a62265 upload the code of MODNet training iteration 2021-01-28 16:57:47 +08:00
Zhanghan Ke ffe384255c
bug fix 2021-01-08 17:50:01 +08:00
ZHKKKe 0bdc3d1ddf upload portrait image/video matting demos of MODNet 2020-12-10 18:36:07 +08:00