Code Monkey home page Code Monkey logo

sam_lora's Introduction

LoRA for SAM (meta's segment-anything)

Usage

from segment_anything import build_sam, SamAutomaticMaskGenerator 
from segment_anything import sam_model_registry
from sam_lora import LoRA_Sam
import torch
sam = sam_model_registry["vit_b"](checkpoint="sam_vit_b_01ec64.pth")
lora_sam = LoRA_Sam(sam,r = 4)
result = lora_sam.sam.image_encoder(torch.rand(size=(1,3,1024,1024)))
print(result.shape)

Train

Coming soon and welcome pull request.

Thanks

The code for LoRA ViT comes form https://github.com/JamesQFreeman/LoRA-ViT

sam_lora's People

Contributors

jamesqfreeman avatar

Stargazers

Danqing Kang avatar Fuqiang Chen avatar QianChen avatar Yang Jin avatar sinimseob avatar Zhao Yan avatar Liu Lanxin avatar avatar ElepherAI avatar Shakeel avatar avatar avatar avatar niuzhuangqun avatar avatar Kyobo avatar flmgjk avatar avatar Adam avatar avatar Black avatar Amir avatar Mehrdad Noori (MeNo) avatar avatar Hung Kao avatar MaLQ avatar Mohammad Reza Taesiri avatar Adeel Ahmad avatar Li Kai avatar Bill Wang avatar Dongjun Hwang avatar avatar Yusin Chen avatar Chongkai Yu avatar Matous Elphick avatar avatar CS-OS avatar Cen WeiXuan avatar ranpin avatar avatar avatar Sudharshan avatar Jonathan Gray avatar avatar Jeffrey Hsu avatar chenshanliang avatar Malik Hashmat avatar avatar avatar JiayiChen815 avatar Alexandre Stenger avatar avatar VeeeeraJiang avatar LeoYuuu avatar Kaikai Tian avatar Fenghe Tang avatar Keyan Zhai avatar Zsj avatar avatar nanase1025 avatar Zhao Chuang avatar SimonTang avatar Yifan Pu avatar jjin avatar Yuseung (Phillip) Lee avatar Zhongyx avatar injo kim avatar avatar Patrick avatar zzj avatar Shiman Li avatar lbjcelsius avatar quektc avatar SunGuys avatar avatar atztao avatar xiawei avatar YongDeuk Seo avatar Bin avatar Chenhui Zhang avatar avatar Starsky0426 avatar hypereikon lab avatar avatar David Marx avatar Yuxiang Lin avatar Jack Liu avatar avatar avatar avatar avatar WadeH avatar Jason Ken Adhinarta avatar Kai Zhou avatar avatar avatar rong avatar avatar Mathieu Nalpon avatar avatar

Watchers

James Cloos avatar avatar 张希 avatar Pritesh Jain avatar avatar

Forkers

fossil-explorer baiti01 kkk222iu gitbenxing shiyongde bebc-mia liu3xing3long gaohuan2015 haorand wh-forker yulinshih keroe jie311 daifeng2016

sam_lora's Issues

Question of training

Hi, I want to use lora fine-tune SAM, and when will you open and submit the training codes?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.