U ‰dAã@sddd„Zdd„ZdS)ÚNCHWFcKsªd|kstdƒ‚|dkr”|r@|r$dnd|d<|jj||f|Ž}q¦|jj||ddg|dd…f|Ž}t|ƒ}|j |d|d¡|d<t|ƒ}n|jj||f|Ž}|S)NZis_testzArgument 'is_test' is requiredrééÚuse_gpu_transformZ_nhwc)ÚAssertionErrorÚnetZ ImageInputÚlistZ NHWC2NCHWÚtuple)ÚmodelÚblob_inÚblob_outÚorderrÚkwargsÚoutputsZ outputs_list©rú?/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/tools.pyÚ image_input s"ÿÿ rcKs|jj||f|Ž}|S)N)rZ VideoInput)r r r r rrrrÚ video_inputsrN)rF)rrrrrrÚ sÿ