在调试DeRPN的时候,训练一段时间就会挂掉。想看看是哪张图引起的崩溃,就想在取minibatch的时候,输出训练的图片名。
方案1:使用print
print "Processing.... ", roidb[i]['image']
方案2:使用logging库
logging.warning(roidb[i]['image'])
方案1的缺点是:输出和GPU训练输出对不上,时常错乱;
方案2可以解决这个问题:
参考链接:
在调试DeRPN的时候,训练一段时间就会挂掉。想看看是哪张图引起的崩溃,就想在取minibatch的时候,输出训练的图片名。
方案1:使用print
print "Processing.... ", roidb[i]['image']
方案2:使用logging库
logging.warning(roidb[i]['image'])
方案1的缺点是:输出和GPU训练输出对不上,时常错乱;
方案2可以解决这个问题:
参考链接: