algoritmo de reconocimiento de imagenes de 2 plagas y 2 enfermedades de la planta de arroz

necesito un algoritmo de reconocimiento de imagen es decir al momento de tomar una imagen de la planta me dira que plaga o enfermedad es, si es posible usando tensorflow en google colad para luego sea exportado a un archivo js