add input train image index