aarohisingla / fastervit Goto Github PK

      This code is tested on python 3.11
      
      python.exe -m pip install --upgrade pip
      
      pip install fastervit>=0.9.8
      
      # Get requirements.txt from here:   https://github.com/NVlabs/FasterViT/blob/main/requirements.txt
      
      pip install -r requirements.txt
      
      pip install matplotlib
      
      pip install opencv-python
      
      # Create a tmp folder where pretrained model will be downloaded.

 from fastervit import create_model
 
 Define fastervit-0 model with 224 x 224 resolution
 
  model = create_model('faster_vit_0_224', 
                           pretrained=True,
                           model_path="/tmp/faster_vit_0.pth.tar")

 from fastervit import create_model
 
  Define fastervit-0 model with 224 x 224 resolution
 
 model = create_model('faster_vit_0_224', 
                           pretrained=True,
                           model_path="/tmp/faster_vit_0.pth.tar")