《原创》mmdetection在训练过程中打印batch中的图像名

在linux下使用find找到epoch_based_runner.py

在train()中找到:

for i, data_batch in enumerate(self.data_loader):
    self._inner_iter = i
    self.call_hook('before_train_iter')
    加入:
    print(data_batch['img_metas'])

 

posted @ 2022-05-31 14:47  海_纳百川  阅读(32)  评论(0编辑  收藏  举报
本站总访问量