subhacom / argos Goto Github PK

I seem to have successfully trained the YOLACT backbone network within Google Colab (since I do not have a GPU). I am now trying to use my trained network within the argos_track utility to analyse some locust videos. However, every time I press play within the GUI, it collapses and no tracking occurs.

Here is the code within the Anaconda Prompt:
(track) C:\Users\Berni>python -m argos_track
Loaded C-utilities with pyximport
Loaded C-function for video capture.
WARNING: CUDA not supported in this installation of PyTorch
INFO:root:Output file "C:/Users/Berni/Videos/2013-08-01_Leics_greg/L.greg-0001-L.mp4.h5"
INFO:root:No segmentation data. Not writing file.
INFO:root:First frame read @ 2022-11-11 09:03:18.462437
Use classes set()
C:\Users\Berni\anaconda3\envs\track\lib\site-packages\torch\jit_recursive.py:238: UserWarning: 'lat_layers' was found in ScriptModule constants, but it is a non-constant submodule. Consider removing it.
warnings.warn("'{}' was found in ScriptModule constants, "
C:\Users\Berni\anaconda3\envs\track\lib\site-packages\torch\jit_recursive.py:238: UserWarning: 'pred_layers' was found in ScriptModule constants, but it is a non-constant submodule. Consider removing it.
warnings.warn("'{}' was found in ScriptModule constants, "
C:\Users\Berni\anaconda3\envs\track\lib\site-packages\torch\jit_recursive.py:238: UserWarning: 'downsample_layers' was found in ScriptModule constants, but it is a non-constant submodule. Consider removing it.
warnings.warn("'{}' was found in ScriptModule constants, "
Starting at first frame
Traceback (most recent call last):
File "C:\Users\Berni\anaconda3\envs\track\lib\site-packages\argos\vwidget.py", line 458, in playVideo
self.timer.start(t)
TypeError: arguments did not match any overloaded call:
start(self, int): argument 1 has unexpected type 'float'
start(self): too many arguments

I can play the video within the GUI but when I click on it, it collapses. I am using a Windows system and assume it is not automatically converting the float into an integer.

The lines of code in which this has occurred:
Traceback (most recent call last):
File "C:\Users\Berni\anaconda3\envs\track\lib\site-packages\argos\frameview.py", line 569, in setRectangles
self._updateItemDisplay()
File "C:\Users\Berni\anaconda3\envs\track\lib\site-packages\argos\frameview.py", line 401, in _updateItemDisplay
pen.setWidth(self.linewidth)
TypeError: setWidth(self, int): argument 1 has unexpected type 'float'

There is no recognition of the locust in my videos as an object, despite having trained a neural network. In looking at the code with my supervisor, we saw there was mention of bbox and were wondering if it has to do with this code above and that's why no bounding box is being displayed. Another possibility we were thinking of is that it is not an error within frameview.py but somewhere else that may not be showing up within Anaconda Prompt.