Mushroom ID requires a lot more than just immediately available visuals. You've gotta see what the cap looks like, the stem, how the stem connects to the cap, the specific characteristics of the gills, the substrate it's growing in, and the spore print (i.e. leave it on a piece of white paper, covered, for a number of hours undisturbed so it drops its spores). And even then it can be tough if the mushroom is abnormal or is decaying at all.
With enough info, I'm sure you could train an ML model to ID mushrooms. But you'd need to give it a lot of info to make a successful ID.