iperov af0b3904fc XSeg trainer: additional dssim loss.

2021-04-15 21:43:47 +04:00

..

Model_Quick96

fix depth_to_space for tf2.4.0. Removing compute_output_shape in leras, because it uses CPU device, which does not support all ops.

2020-12-11 11:28:33 +04:00

Model_SAEHD

upd pretrain option help

2021-03-23 16:07:50 +04:00

Model_XSeg

XSeg trainer: additional dssim loss.

2021-04-15 21:43:47 +04:00

__init__.py

DFL-2.0 initial branch commit

2020-01-21 18:43:39 +04:00

ModelBase.py

SAEHD: random flip replaced with random SRC flip(default False) and random DST flip(default True)

2021-03-23 15:13:10 +04:00