آموزش Faster RCNN

با سلام سپاس فراوان از استاد گرامی جهت آموزش بسیار خوب شبکه faster_rcnn
سوال: خطای این قسمت از کد را چطور میشه برطرف نمود؟ باکس های اشیا رو رسم نمیکند.
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

np_arr = tensor.permute(1, 2, 0).numpy()
plt.imshow(np_arr)

ax = plt.gca()
for box, label, score in zip(boxes, labels, scores):
if score > 0.3:
rect = Rectangle((box[0], box[1]),
(box[2] – box[0]),
(box[3] – box[1]),
fill=False,
edgecolor=(1, 0, 0),
linewidth=2)
ax.add_patch(rect)

plt.show()

TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

سلام. المکانش هست بسته اموزشی مشابه این برای mask r-cnn همراه با کدها قرار دهید. ممنونم

سلام اگر امکانش هست با کدنویسی کراس هم الگوریتم faster Rcnn آموزش بدهید ممنون