加上参数vis进行结果可视化出现问题

[wangyf0928]

Load checkpoint from pretrained/13_Faster_RCNN.pt.
[INFO] 2023-07-06 15:43:05 ==================Configs==================
[INFO] 2023-07-06 15:43:05 MODEL:
[INFO] 2023-07-06 15:43:05 NAME: Faster_RCNN
[INFO] 2023-07-06 15:43:05 BACKBONE: resnet50
[INFO] 2023-07-06 15:43:05
[INFO] 2023-07-06 15:43:05 DATA:
[INFO] 2023-07-06 15:43:05 DATASET: mydata
[INFO] 2023-07-06 15:43:05 TRANSFORM: frcnn
[INFO] 2023-07-06 15:43:05 SCALE: [800, 1333]
[INFO] 2023-07-06 15:43:05 OPTIMIZE:
[INFO] 2023-07-06 15:43:05 OPTIMIZER: sgd
[INFO] 2023-07-06 15:43:05 BASE_LR: 0.001
[INFO] 2023-07-06 15:43:05 SCHEDULER: 1x
[INFO] 2023-07-06 15:43:05 BATCH_SIZE: 2
[INFO] 2023-07-06 15:43:05 TEST:
[INFO] 2023-07-06 15:43:05 NMS_THRESH : 0.5
[INFO] 2023-07-06 15:43:05 CONF_THRESH: 0.05
[INFO] 2023-07-06 15:43:05 MISC:
[INFO] 2023-07-06 15:43:05 VAL_FREQ: 1
[INFO] 2023-07-06 15:43:05 SAVE_FREQ: 1
[INFO] 2023-07-06 15:43:05 NUM_WORKERS: 0
[INFO] 2023-07-06 15:43:05 ==================Options==================
[INFO] 2023-07-06 15:43:05 config=configs/faster_rcnn_mydata.yml
[INFO] 2023-07-06 15:43:05 debug=False
[INFO] 2023-07-06 15:43:05 device=cuda:0
[INFO] 2023-07-06 15:43:05 epochs=13
[INFO] 2023-07-06 15:43:05 gpu_id=0
[INFO] 2023-07-06 15:43:05 load=pretrained/13_Faster_RCNN.pt
[INFO] 2023-07-06 15:43:05 local_rank=None
[INFO] 2023-07-06 15:43:05 no_val=False
[INFO] 2023-07-06 15:43:05 resume=False
[INFO] 2023-07-06 15:43:05 save_path=None
[INFO] 2023-07-06 15:43:05 tag=faster_rcnn_mydata
[INFO] 2023-07-06 15:43:05 vis=True
[INFO] 2023-07-06 15:43:05 ===========================================
[INFO] 2023-07-06 15:43:05 ===========================================
[INFO] 2023-07-06 15:43:05 val_trasforms: Compose([
ToTensorV2(always_apply=True, p=1.0, transpose_mask=False),
], p=1.0, bbox_params={'format': 'pascal_voc', 'label_fields': ['labels'], 'min_area': 0, 'min_visibility': 0, 'check_each_transform': True}, keypoint_params=None, additional_targets={})
[INFO] 2023-07-06 15:43:05 ===========================================
Traceback (most recent call last):...........] ETA: 33m
File "eval.py", line 59, in
evaluate(model, val_dataloader, which_epoch, writer, logger, 'val')
File "eval.py", line 20, in evaluate
loss = model.evaluate(dataloader, epoch, writer, logger, data_name)
File "D:\easy_detection\easy_detection\network\Faster_RCNN\Model.py", line 183, in evaluate
return self.eval_mAP(dataloader, epoch, writer, logger, data_name)
File "D:\easy_detection\easy_detection\network\base_model.py", line 95, in eval_mAP
img = tensor2im(image).copy()
File "D:\Anaconda\Anaconda3\envs\python37\lib\site-packages\mscv\image\image_io.py", line 14, in tensor2im
x = x.data.cpu().numpy()[0]
AttributeError: 'list' object has no attribute 'data'