怎么单独去做测试？有单独的test.py文件吗

[long123524]

怎么单独去做测试一些照片？

[zhangxiao121]

您好，这个问题您解决了吗？怎么单独去测试一些图片？

[hahaha1230]

我自己写了个单独测试的代码

import models if name == 'main': resume = 'RCFcheckpoint_epoch12.pth' model = models.RCF() model.eval() checkpoint = torch.load(resume,map_location='cpu') try: model.load_state_dict(checkpoint) except Exception: model.load_state_dict(checkpoint['state_dict'],False) img_path = r'D:\Work\line_extraction\RCF-pytorch-master\BSR\BSDS500\data\images\train\12003.jpg' result_path = './examples/demo.png'

original_img = np.array(cv2.imread(img_path), dtype=np.float32)
h, w, _ = original_img.shape

img = prepare_image_cv2(original_img)
img = torch.from_numpy(img).unsqueeze(0)

outs = model(img)
result = outs[-1].squeeze().detach().cpu().numpy()

result = (result * 255).astype(np.uint8)
Image.fromarray(result).save(result_path)