Open maksim-dh opened 1 month ago
Could you make sure that the environment variables are correctly set as described in this section? https://github.com/rdroste/unisal?tab=readme-ov-file#data
Are these required even when using the pretrained model? Thank you.
I keep encountering the following error on an Ubuntu VM:
torch device set to: cpu Train ID: pretrained_unisal /home/azureuser/miniconda3/envs/unisal/lib/python3.7/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3190.) return _VF.meshgrid(tensors, *kwargs) # type: ignore[attr-defined] Best weights loaded Traceback (most recent call last): File "run.py", line 97, in
fire.Fire()
File "/home/azureuser/miniconda3/envs/unisal/lib/python3.7/site-packages/fire/core.py", line 138, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "/home/azureuser/miniconda3/envs/unisal/lib/python3.7/site-packages/fire/core.py", line 471, in _Fire
target=component.name)
File "/home/azureuser/miniconda3/envs/unisal/lib/python3.7/site-packages/fire/core.py", line 675, in _CallAndUpdateTrace
component = fn( varargs, **kwargs)
File "run.py", line 70, in predictions_from_folder
folder_path, is_video, source=source, model_domain=model_domain)
File "/home/azureuser/unisal/unisal/train.py", line 935, in generate_predictions_from_path
images_path, source=source, frame_modulo=frame_modulo
File "/home/azureuser/unisal/unisal/data.py", line 1123, in init
img = cv2.imread(str(frame_files[0]))
IndexError: list index out of range
I have all video frames downloaded as png files in a separated folder ~/videos/frames/
The command I am using to try and activate the pretrained model is this :~/unisal$ python run.py predictions_from_folder ~/videos/frames --is_video True --train_id pretrained_unisal
Would appreciate any possible assistance with this, thank you!