This isn't exactly what you asked for, but there's a "drumsep" model, which takes a drum audio track and separates it into 6 stems: kick, snare, toms, hi-hat, ride, and crash.
I’m the author of the high resolution guitar model posted in a comment above. I have a drum transcription model that I’m getting ready for release soon which should be state of the art for this. I’ll try to update this thread when I’m done
Ctrl+F for "drumsep" in this doc:
Instrumental, vocal & other stems separation & mix/master guide - UVR/MDX/Demucs/GSEP & others - Google Docs https://docs.google.com/document/d/17fjNvJzj8ZGSer7c7OFe_CNf...