测试数据

[oky921]

你好，我现在想用你的这个测试代码检测一个数据，然后这个数据是我自己找的一个，如果想要在你这个上面测试，我该对数据进行怎样的调整呢？

[skJack]

您好 谢谢关注，可以参考看test.py中的这个部分：
hint_source = datas['hint'].cuda()

feature = model.control_model.input_hint_block.forward_features(hint_source) #[bs,320,32,32]
output = model.control_model.global_pool(feature).flatten(1)
cls_output = model.control_model.fc(output)

把hint_source改成你的数据就行