o
    *j0                     @   sF   d dl mZ d dlmZ d dlmZ ejejdG dd deZdS )    )Trainers)TRAINERS)EpochBasedTrainer)module_namec                       sD   e Zd Z fddZdd Z fddZ fddZd	d
 Z  ZS ) ImageInstanceSegmentationTrainerc                       t  j|i | d S N)super__init__selfargskwargs	__class__ {/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/trainers/cv/image_instance_segmentation_trainer.pyr
   
      z)ImageInstanceSegmentationTrainer.__init__c                 C   s   |S r   r   )r   datar   r   r   
collate_fn   s   z+ImageInstanceSegmentationTrainer.collate_fnc                    r   r   )r	   trainr   r   r   r   r      r   z&ImageInstanceSegmentationTrainer.trainc                    s   t  j|i |}|S r   )r	   evaluate)r   r   r   Zmetric_valuesr   r   r   r      s   z)ImageInstanceSegmentationTrainer.evaluatec                 C   s   d S r   r   )r   modelZinputsr   r   r   prediction_step   s   z0ImageInstanceSegmentationTrainer.prediction_step)	__name__
__module____qualname__r
   r   r   r   r   __classcell__r   r   r   r   r      s    r   N)	Zmodelscope.metainfor   Zmodelscope.trainers.builderr   Zmodelscope.trainers.trainerr   Zregister_moduleZimage_instance_segmentationr   r   r   r   r   <module>   s
   