python assert
assert 作用为:判断一个表达式是否正确,若返回值为false,触发异常,并输出自己设置的提示信息。
style_img = image_loader("data/starry_night.jpg")
content_img = image_loader("data/shanghai_buildings.bmp")
print("style size:", style_img.size()) # style size: torch.Size([1, 3, 128, 600])
print("content size:", content_img.size()) # content size: torch.Size([1, 4, 128, 600])
assert style_img.size() == content_img.size(), "we need to import style and content images of the same size"
输出为: