为什么在Pytorch中print语句有时无法打印?

e0uiprwp  于 12个月前  发布在  其他
关注(0)|答案(2)|浏览(147)

我正在使用pycocotools库,以及DETR(用于对象检测的可变形变换器)的实现。
我正在调试代码的一部分,它调用了loadRes函数[这里][1]。
我在res = COCO()行之前和之后添加了print语句,因为res.dataset['images'] = [img for img in self.dataset['images']]行给出了一个错误(AttributeError: 'NoneType' object has no attribute 'dataset')。
当我试图跟踪代码并找到res的当前值时(特别是,为什么它是 None),我注意到我的print语句没有出现在输出中,尽管在回溯中明显存在。
为什么会发生这种情况?

AttributeError                            Traceback (most recent call last)
<ipython-input-4-ff0c7180ea84> in <module>
    175 
--> 176     test_stats, coco_evaluator = evaluate(
    177         model, criterion, postprocessors, data_loader_val, base_ds, "cuda", args["output_dir"]
    178     )

~/anaconda3/lib/python3.8/site-packages/torch/autograd/grad_mode.py in decorate_context(*args, **kwargs)
     25         def decorate_context(*args, **kwargs):
     26             with self.clone():
---> 27                 return func(*args, **kwargs)
     28         return cast(F, decorate_context)
     29 

~/Desktop/traffic_sign_relevance-main/attention/engine.py in evaluate(model, criterion, postprocessors, data_loader, base_ds, device, output_dir)
    128         res = {target['image_id'].item(): output for target, output in zip(targets, results)}
    129         if coco_evaluator is not None:
--> 130             coco_evaluator.update(res)
    131 
    132         if panoptic_evaluator is not None:

~/Desktop/traffic_sign_relevance-main/attention/datasets/coco_eval.py in update(self, predictions)
     52             with open(os.devnull, 'w') as devnull:
     53                 with contextlib.redirect_stdout(devnull):
---> 54                     coco_dt = COCO.loadRes(self.coco_gt, results) if results else COCO()
     55             coco_eval = self.coco_eval[iou_type]
     56 

~/anaconda3/lib/python3.8/site-packages/pycocotools/coco.py in loadRes(self, resFile)
    316         print("In loadRes")
    317         print(type(res))
--> 318         res.dataset['images'] = [img for img in self.dataset['images']]
    319 
    320         print('Loading and preparing results...')

AttributeError: 'NoneType' object has no attribute 'dataset' ```



  [1]: https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocotools/coco.py

字符串

zpf6vheq

zpf6vheq1#

这是因为pycocotools打印的第52、53行。

51             """
 52             with open(os.devnull, 'w') as devnull:
 53                 with contextlib.redirect_stdout(devnull):
 54                     coco_dt = COCO.loadRes(self.coco_gt, results) if results else COCO() 
 55             """
 56             coco_dt = COCO.loadRes(self.coco_gt, results) if results else COCO()
 57             coco_eval = self.coco_eval[iou_type]

字符串
您的打印报表将显示在屏幕上。

dsekswqp

dsekswqp2#

也许你的notebook阻止了一个已安装包中的print语句?

相关问题