I have done the two steps 'Performing OSHOT pretraining' and 'Testing pretrained model', however get this error when running 'Performing OSHOT adaptation':
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
Traceback (most recent call last):
File "tools/oshot_net.py", line 119, in
main()
File "tools/oshot_net.py", line 108, in main
cfg=cfg
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/engine/oshot.py", line 151, in oshot_inference
predictions = compute_on_dataset(model, data_loader, device, oshot_breakpoints, inference_timer, cfg)
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/engine/oshot.py", line 99, in compute_on_dataset
log_test_image(cfg, summary_writer, "detections_{}_its".format(oshot_it+1), image_ids[0], images, output, image_name=image_name)
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/utils/log_image_bb.py", line 136, in log_test_image
log_image_and_bb(summary_writer, name, global_step, image, targets, image_name)
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/utils/log_image_bb.py", line 32, in log_image_and_bb
image = _overlay_boxes(image, targets)
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/utils/log_image_bb.py", line 98, in _overlay_boxes
colors = _compute_colors_for_labels(labels).tolist()
File "/home/hl/hl/oshot_detection-master/maskrcnn_benchmark/utils/log_image_bb.py", line 71, in _compute_colors_for_labels
cmcolors.append(np.array(cmap(lbl)[:3])*255)
File "/home/hl/anaconda3/envs/python37/lib/python3.7/site-packages/matplotlib/colors.py", line 561, in call
elif np.any(np.isnan(X)):
File "<array_function internals>", line 6, in any
File "/home/hl/anaconda3/envs/python37/lib/python3.7/site-packages/numpy/core/fromnumeric.py", line 2330, in any
return _wrapreduction(a, np.logical_or, 'any', axis, None, out, keepdims=keepdims)
File "/home/hl/anaconda3/envs/python37/lib/python3.7/site-packages/numpy/core/fromnumeric.py", line 85, in _wrapreduction
return reduction(axis=axis, out=out, **passkwargs)
TypeError: any() received an invalid combination of arguments - got (out=NoneType, axis=NoneType, ), but expected one of:
- ()
- (name dim, bool keepdim)
didn't match because some of the keywords were incorrect: out, axis
- (int dim, bool keepdim)
didn't match because some of the keywords were incorrect: out, axis
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////