创建简单的几何形状 img = torch.zeros(1, size, size) # 添加随机形状 x = np.random.randint(5, size -5) y = np.random.randint(5, size-5) r = np.random.randint(3, 8) cv2.circle(img