claudecode/AvianVisitors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

server.py add support for Data_Model_V2

Browse Source
This commit is contained in:
frederik
2024-02-19 14:39:25 +01:00
parent b4e8fee59e
commit 31e1379f14
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/server.py
+6 -1
View File
@@ -72,8 +72,13 @@ def loadMetaModel():
global M_INPUT_LAYER_INDEX global M_INPUT_LAYER_INDEX
global M_OUTPUT_LAYER_INDEX global M_OUTPUT_LAYER_INDEX
if get_settings().getint('DATA_MODEL_VERSION') == 2:
data_model = 'BirdNET_GLOBAL_6K_V2.4_MData_Model_V2_FP16.tflite'
else:
data_model = 'BirdNET_GLOBAL_6K_V2.4_MData_Model_FP16.tflite'
# Load TFLite model and allocate tensors. # Load TFLite model and allocate tensors.
M_INTERPRETER = tflite.Interpreter(model_path=userDir + '/BirdNET-Pi/model/BirdNET_GLOBAL_6K_V2.4_MData_Model_FP16.tflite') M_INTERPRETER = tflite.Interpreter(model_path=os.path.join(userDir, 'BirdNET-Pi/model', data_model))
M_INTERPRETER.allocate_tensors() M_INTERPRETER.allocate_tensors()
# Get input and output tensors. # Get input and output tensors.