9i~bSSKJr SSKrSSKJrJr SSKJrJ r J r J r SSK J r "SS\5rg) )DictN)ModelTensor) content_loss guided_filter style_losstotal_variation_loss)unet_generatorc ^\rSrSrU4SjrS S\\\4S\\\4S\\\4S\\\44SjjrS\\\4S\\\44S jr S r U=r $) CartoonModel c:>[TU]"U/UQ70UD6 Xlg)N)super__init__ model_dir)selfrargskwargs __class__s e/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/cv/cartoon/model_tf.pyrCartoonModel.__init__s 4T4V4" input_photo input_cartooninput_superpixelreturncUc[U5n[XSS9nSU0$[U5n[XSS9n[URUXS5n[ X%5upx[ U5n SU-SU--SU --n Un UU U S.$)areturn the result by the model Args: input_photo: the preprocessed input photo image input_cartoon: the preprocessed input cartoon image input_superpixel: the computed input superpixel image Returns: output_dict: output dict of target ids r)routput_cartoong?gi@g@)r g_lossd_loss)r rrrr r ) rrrroutputr con_loss sty_g_loss sty_d_losstv_lossr!r"s r__call__CartoonModel.__call__s  #K0F*;!DN$n5 5#K0F*;!DN#DNNK$2FH%/ %N "J*>:GJ&x7#-GFF#1   rinputcg)z Run the forward pass for a model. Args: input (Dict[str, Tensor]): the dict of the model inputs for the forward method Returns: Dict[str, Tensor]: output from the model forward pass N)rr*s rforwardCartoonModel.forward7sr)r)NN) __name__ __module__ __qualname____firstlineno__rrstrrr(r-__static_attributes__ __classcell__)rs@rr r s#0426 #c6k*# V ,##3;/ #<@V ;L #J  T#v+.  4V 3D   rr )typingr tensorflowtfmodelscope.models.baserrlossrrr r networkr r r,rrr<s%0OO#4 54 r