mirror of
https://github.com/Ascend/ascend_community_projects.git
synced 2025-09-26 20:01:17 +08:00
update UAV_CrowdCounting/main/main.py.
Signed-off-by: koervcor <1015296415@qq.com>
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user