change_detection.pytorch
change_detection.pytorch copied to clipboard
缺少单独调用模型用于测试的脚本
可以单独提供一个用于测试新的两张输入图片的脚本吗?
我已经训练好了模型,并且在最后一步infer_vis出来的验证图片效果不错
但是缺乏一个可以单独调用模型用于测试的脚本,请问作者有什么解决办法或者可以提供一下代码吗
我也遇到了同样的问题, 代码中有给到: """ Note: if you use sliding window inference, set: from change_detection_pytorch.datasets.transforms.albu import ( ChunkImage, ToTensorTest)
test_transform = A.Compose([
A.Normalize(),
ChunkImage({window_size}}),
ToTensorTest(),
], additional_targets={'image_2': 'image'})
""" 其中的A 具体是指什么
@xiaxialin A指的是albumentations
@likyoo 有测试脚本提供吗?或者思路