diff --git a/Inject_and_Visualize.py b/Inject_and_Visualize.py index 404b0c40f688dc1179851f76d453b8f0f122cbdb..34bcdc83745be7a39ec3598cc9fcd32f43fc88b9 100644 --- a/Inject_and_Visualize.py +++ b/Inject_and_Visualize.py @@ -18,10 +18,7 @@ from hamer.utils.extraction_utils import decompress_from_lzma, get_video_fps, pr LIGHT_BLUE=(0.65098039, 0.74117647, 0.85882353) from vitpose_model import ViTPoseModel -from typing import Dict, Optional -import subprocess import gc -import lmdb, pickle, lzma def main(args, model, device): initial_start_time = time.time() diff --git a/README.md b/README.md index c0dc12aec0d5a69b8284f8622365b6fac66f3c2b..754f35cd310939e71e8185eb9a6b10f43d669e98 100644 --- a/README.md +++ b/README.md @@ -9,8 +9,10 @@ This Repository is a modified version of the original HaMeR implementation, and The following guide is written based on the modifications made to HaMeR. To refer to the original guide, follow this [link](https://github.com/geopavlakos/hamer). ### Environment +*Note:* There might be some issues with the variables of `DEFAULT_CHECKPOINT` and `CACHE_DIR_HAMER`, as they are currently checkpoints stored in my user space. + #### Container Registry -HaMeR can be problematic to set up. The easy way out is to run it on the Surrey Cluster. To do this, you can simply go to the [container registry](https://container-registry.surrey.ac.uk/harbor/projects/2/repositories) and look in the `shared container` section where all the containers are stored. The container needed for this repository is named `hamer`. +HaMeR can be problematic to set up. The easy way out is to run it on the Surrey Cluster. To do this, you can simply go to the [container registry](https://container-registry.surrey.ac.uk/harbor/projects/2/repositories) and look in the `shared container` section where all the containers are stored. The container needed for this repository is named `hamer`. An example `submit_file.sh` is provided for reference. #### Local The same exact Dockerfile in the registry has been uploaded to the docker folder of this repo. Note that this Dockerfile has been modified from the original, but it seems to work well. diff --git a/submit_file.sh b/submit_file.sh new file mode 100644 index 0000000000000000000000000000000000000000..7e4a37aa14ba61459fff8ffe703bdba9d2d43c5e --- /dev/null +++ b/submit_file.sh @@ -0,0 +1,81 @@ +JobBatchName = "Run_HaMeR" +# ---------------------------------------------------------------------------------------------------------------------- +executable=/bin/bash +# ---------------------------------------------------------------------------------------------------------------------- +universe=docker +docker_image=container-registry.surrey.ac.uk/shared-containers/hamer:SHA_cb1639dd +should_transfer_files=YES +environment="mount=/vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release" +# ---------------------------------------------------------------------------------------------------------------------- +log = /mnt/fast/nobackup/scratch4weeks/$(cluster).$(process).log +output = /mnt/fast/nobackup/scratch4weeks/$(cluster).$(process).out +error = /mnt/fast/nobackup/scratch4weeks/$(cluster).$(process).err +# ---------------------------------------------------------------------------------------------------------------------- +Requirements = (machine != "elenwe.eps.surrey.ac.uk") && ( HasStornext == true ) +# ---------------------------------------------------------------------------------------------------------------------- +request_CPUs = 1 +request_GPUs = 1 +# ++GPUMem = 10000 +request_memory = 20G +# ---------------------------------------------------------------------------------------------------------------------- ++CanCheckpoint = False ++JobRunTime = 1 +# ---------------------------------------------------------------------------------------------------------------------- +on_exit_hold = ExitCode != 0 +# ---------------------------------------------------------------------------------------------------------------------- +launcher = $ENV(PWD)/launch.sh +# ---------------------------------------------------------------------------------------------------------------------- +repository = https://gitlab.surrey.ac.uk/JianHe/hamer.git +repo_name = Hamer_Repo +# ---------------------------------------------------------------------------------------------------------------------- +branch = HEAD +# ---------------------------------------------------------------------------------------------------------------------- +python = python3 +# ---------------------------------------------------------------------------------------------------------------------- +arguments = $(launcher) $(repo_name) $(repository) $(branch) $(python) $(args) +# ---------------------------------------------------------------------------------------------------------------------- +args = $(script) +# ---------------------------------------------------------------------------------------------------------------------- +script = Extract_HaMeR_Single.py --vid $(Input_Path) --out_folder $(Output_Path) + +# ---------------------------------------------------------------------------------------------------------------------- + +Output_Path = /mnt/fast/nobackup/scratch4weeks/jl02958/BOBSL_HaMeR_Extras/ + +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6247230507285724422.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5408597340544357103.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6784596629031752949.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6196771943002142813.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5620842599234468139.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5582619536953498553.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5213762162115483713.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5296891683626303285.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6270974804484919888.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6335904402078950920.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6120093891887671821.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6075933467625912531.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5595214528548999641.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5987513405399935912.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5352897198172711984.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/6243537694404620461.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5212610251886695418.masked.mp4 +queue 1 +Input_Path = /vol/research/LF_datasets/mixedmode/BOBSL/bobsl-v1-release/MASKED_VIDEOS/5603759795481130757.masked.mp4 +queue 1 \ No newline at end of file