PGD对抗训练读取数据使用的ImageNetTrainPipeV2问题

[magic-liu2021]

   在进行PGD对抗训练时，需要读取数据，而这用到了ImageNetTrainPipeV2，我无法成功运行prototype/data/pipelines/imagenet_pipeline_v2.py，
   在以下代码中：
   import nvidia.dali.ops as ops
   self.mc_input = ops.McReader(file_root=data_root,
                                                       file_list=data_list,
                                                       sampler_index=list(sampler))        
   在nvidia.dali.ops中没有McReader方法，这部分代码是如何实现的？        
   感谢您的帮助。

[PanXiebit]

   在进行PGD对抗训练时，需要读取数据，而这用到了ImageNetTrainPipeV2，我无法成功运行prototype/data/pipelines/imagenet_pipeline_v2.py，
   在以下代码中：
   import nvidia.dali.ops as ops
   self.mc_input = ops.McReader(file_root=data_root,
                                                       file_list=data_list,
                                                       sampler_index=list(sampler))        
   在nvidia.dali.ops中没有McReader方法，这部分代码是如何实现的？        
   感谢您的帮助。

请问您解决这个问题了吗？

[magic-liu2021]

   在进行PGD对抗训练时，需要读取数据，而这用到了ImageNetTrainPipeV2，我无法成功运行prototype/data/pipelines/imagenet_pipeline_v2.py，
   在以下代码中：
   import nvidia.dali.ops as ops
   self.mc_input = ops.McReader(file_root=data_root,
                                                       file_list=data_list,
                                                       sampler_index=list(sampler))        
   在nvidia.dali.ops中没有McReader方法，这部分代码是如何实现的？        
   感谢您的帮助。

请问您解决这个问题了吗？

没解决呀，据说这是他们内部的实现方法。可以换成PIL读取图像。