A census of $\rho$ Oph candidate members from Gaia DR2