A.- Par décision du 26 juin 1999, l'Office de l'assurance-invalidité du canton du Valais (ci-après : OAI) a mis A._ au bénéfice d'une rente entière d'invalidité, fondée sur un taux d'invalidité de 100 %, avec effet rétroactif au 1er août 1998. A l'issue d'une procédure de révision du droit à la rente, se fondant sur un rapport d'expertise du docteur B._, spécialiste en médecine physique, réhabilitation et rhumatologie - daté par inadvertance de mai 2000, mais reçu le 21 juillet 2000 - l'OAI a, par décision du 18 août 2000, supprimé la rente entière d'invalidité à partir du 1er octobre 2000. Il a également classé le dossier en ce qui concernait les mesures d'ordre professionnel, en indiquant à l'assuré qu'il lui était loisible de déposer une nouvelle demande tendant à l'octroi de cette prestation, s'il manifestait le désir de collaborer activement à la recherche d'une solution professionnelle. B.- Saisi d'un recours de A._, le Tribunal des assurances du canton du Valais l'a rejeté par jugement du 26 février 2001, en renvoyant le dossier à l'OAI pour qu'il examine la (nouvelle) demande d'aide au placement formée devant lui. C.- A._ interjette recours de droit administratif contre ce jugement, dont il demande l'annulation, en concluant à l'octroi d'une demi-rente d'invalidité. Il s'appuie sur un rapport du docteur C._, médecin traitant, dont il ressort que son incapacité de travail actuelle serait de 50 %. L'OAI propose le rejet du recours. L'Office fédéral des assurances sociales ne s'est pas déterminé. D.- Par décision sur opposition du 13 mars 2000, la CNA a confirmé la décision du 18 novembre 1999 par laquelle elle a alloué à A._ une rente d'invalidité de 25 %, à partir du 1er novembre 1999. 