@atyadav you have only 53 items, I will recommend to increase images and introduce 3rd label images as โnoneโ images which contain images which are not โokโ and โstopโ images and will act as background subtractors or in simpler terms will increase accuracy in transfer learning.