-
Notifications
You must be signed in to change notification settings - Fork 8
Open
Description
Step 1
git clone https://github.com/HowieMa/CVTHead/
Step 2
cd CVTHead
Step 3
conda create -n cvthead python=3.9
Step 4
conda activate cvthead
Step 5
Download and replace the requirements.txt
requirements.txt
Step 6
pip install -r requirements.txt
Step 7
pip install torch-scatter -f https://data.pyg.org/whl/torch-2.0.0+cu118.html
Step 8
Download all the models from the following links and place in data/ folder. You need to register before you could download Flame2020.zip
https://docs.google.com/uc?export=download&id=1CJL03b7kO45Pz9iG17gnlNJThXocD5Mt
https://docs.google.com/uc?export=download&id=17bGCDQLuXU81xqHF1MB6nBqpBO6PtPd2
https://docs.google.com/uc?export=download&id=1rp8kdyLPvErw2dTmqtjISRVvQLj6Yzje
https://docs.google.com/uc?export=download&id=154JgKpzCPW82qINcVieuPH3fZ2e0P812
https://docs.google.com/uc?export=download&id=1rLtc037Ra6Z6t0kp-gJ8P1ZKfzkKm070
https://docs.google.com/uc?export=download&id=1Enw9MU9Xin77ws08y4pNqkMW0AyUIzv_
https://download.is.tue.mpg.de/download.php?domain=flame&sfile=FLAME2020.zip&resume=1
After downloading Flame2020.zip, extract generic_model.pkl and place in data folder.
Step 9
Refer the Inference commands mentioned in home page
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels