ReflectionEraser/ReflectionEraserApp
0
1U
2 � d� � @ s� d dl Z d dlT d dlmZ d dlmZ d dlmZ d dlm Z d dl3m Z d dl m
Z
d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ed�dd�ZdS )� N)�*)�CosineLR)�namec C s2 t �d| �}| �d�r$t|d�}n4t|d�}|S )Nzmodels.ZtwostageZTwoStageYTMTNetModelZ YTMTNetModel)� importlib�
import_module�5startswith�getattr)r �base�model� r �9D:\workspace\python\YTMT-Strategy-main\models\__init__.py�6make_model s7 8 9r
) r Z models.archZmodels.lr_schedulerr Zmodels.losses_r2Z losses_r2Zmodels.losses_r2bZ10losses_r2bZmodels.losses_r2b2Z losses_r2b2Zmodels.losses_r2_simpleZlosses_r2_simpleZ models.losses_r2_simple_l1ssimZlosses_r2_simple_l1ssimZ"models.losses_r2_simple_l1ssim_decZlosses_r2_simple_l1ssim_decZ models.losses_r2_simple_ssimZlosses_r2_simple_ssimZ models.losses_r2_simple_ssim_2Zlosses_r2_simple_ssim_2Zmodels.losses_r2_ablation4Zlosses_r2_ablation4Zmodels.losses_r2_ablation5Zlosses_r2_ablation5Zmodels.losses_r2_ablation14Zlosses_r2_ablation14�strr
r r r r �<module> s