search

YttriLab / A-SOID

An active learning platform for expert-guided, data efficient discovery of behavior.
Other
55 stars 7 forks source link

Error during 'Upload Data' after excluding keypoints #73

Closed Mijar007 closed 8 months ago

Mijar007 commented 8 months ago

Describe the bug My pose files have 12 bodyparts. I deselected four of them and pushed the 'Create Project/Preprocess' button. This rose the following error:

IndexError: index 8 is out of bounds for axis 1 with size 8
Traceback:
File "C:\ProgramData\anaconda3\envs\asoid\lib\site-packages\streamlit\runtime\scriptrunner\script_runner.py", line 535, in _run_script
    exec(code, module.__dict__)
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\app.py", line 328, in <module>
    main()
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\app.py", line 292, in main
    A_data_preprocess.main(config=st.session_state['config'])
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\apps\A_data_preprocess.py", line 34, in main
    processor.main()
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\utils\load_preprocess.py", line 578, in main
    self.compile_data()
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\utils\load_preprocess.py", line 485, in compile_data
    filt_pose, _ = adp_filt(current_pose, idx_selected, idx_llh, self.llh_value)
File "C:\Users\Rabenstein\Python\A-SOID-main\asoid\utils\preprocessing.py", line 65, in adp_filt
    currdf_filt[0, (2 * x):(2 * x + 2)] = np.hstack([datax[0, x], datay[0, x]])

If I include the deselected bodyparts afterwards and push the button again, the error will not rise.

Expected behavior No error and the creation of a dataset without the deselected bodyparts.

Screenshots image

Desktop (please complete the following information): Win 11 pro, 10.0.22631, x64, German language

packages in environment at C:\ProgramData\anaconda3: Name Version Build Channel _anaconda_depends 2023.09 py311_mkl_1 anaconda-anon-usage 0.4.2 py311hfc23b7f_0 anaconda-catalogs 0.2.0 py311haa95532_0 anaconda-client 1.12.1 py311haa95532_0 anaconda-cloud-auth 0.1.4 py311haa95532_0 anaconda-navigator 2.5.0 py311haa95532_0 anaconda-project 0.11.1 py311haa95532_0

General Python 3.11.5 In Asoid environment: Python 3.8.18

MS Edge used as Browser

Project Config (please post the content of the corresponding config.ini file) Will not be generated because the error rises before it is generated.

Additional context Add any other context about the problem here.

Best wishes, Michael

JensBlack commented 8 months ago

this is likely related to #70 . Thank you for finding it.

I've been testing with SLEAP and OpenMonkeyStudio pose data, so the culprit must be the filter step we do for DLC data.

I will work on fixing.