update UAV_CrowdCounting/main/main.py.

Signed-off-by: koervcor <1015296415@qq.com>
This commit is contained in:
koervcor
2022-10-24 05:40:48 +00:00
committed by Gitee
parent c03b749a53
commit d14f668963

View File

@@ -44,12 +44,9 @@ if __name__ == '__main__':
with open("./data/test.jpg", 'rb') as f:
dataInput.data = f.read()
TEST_NAME = "'./data/test.jpg'"
elif os.path.exists('./data/test.png') == 1:
with open("./data/test.png", 'rb') as f:
dataInput.data = f.read()
TEST_NAME = "'./data/test.png'"
else:
print("The test image does not exist.")
exit()
STREAM_NAME = b'uav_crowdcounting'
IN_PLUGIN_ID = 0