BiRefNet icon indicating copy to clipboard operation
BiRefNet copied to clipboard

Make it easier to know which datasets were used

Open set-soft opened this issue 2 months ago • 2 comments

Hi! The availability of various pre-trained weights is very good, thanks! You also include a detailed information about the datasets in the "model zoo" section. Thanks again. But then I get confused about the relation between the models available in the Google Drive and the models uploaded on Hugging Face. Can you please add a link to Hugging Face along with the Google Drive link?

BTW: The model cards on Hugging Face have too much copy & paste, I suggest putting all the repetitive instructions in just one repo (https://huggingface.co/ZhengPeng7/BiRefNet) and in the other repos just put information relevant to the weights found in the repo, with links to the main HF repo, paper, GitHub, etc.

I wrote some ComfyUI nodes that supports all the pre-trained models you uploaded to Hugging Face and want to make a table indicating which data set was used for each case.

P.S. I downloaded the General HR (epoch 230) thinking it was the same as https://huggingface.co/ZhengPeng7/BiRefNet_HR/resolve/main/model.safetensors but found this isn't the case. I don't mean the file format, I mean a SHA256 for the bb.layers.0.blocks.0.attn.proj.bias Tensor. This is why I ask for more detailed information, you might also mention which epoch is the one on Hugging Face, so we know if the model from other of your sources is newer.

set-soft avatar Oct 24 '25 16:10 set-soft

Oh, BTW! Adding this kind of information to the safetensors metadata will be great! (dataset used, epoch, URL, dtype, etc.)

set-soft avatar Oct 24 '25 16:10 set-soft

Oh, my bad... Thanks a lot, @set-soft. I have read your long and useful msg days ago. But I've been too busy for the dentist and flight to my phd school. What you wrote is also what i wanted. But I was also first to set up such many models as a sery. I know the problem but was confused to solve it. Now, thanks to your suggestions, and I also thought about it. I will spare time to make them up. If not suitable due to the compatibility, I'll remember to upgrade the criteria in the v2 version Anyway, thanks again :)

ZhengPeng7 avatar Oct 30 '25 13:10 ZhengPeng7