增加了相似菜品预览功能，更完整了，交互更方便了。

train/train_embedding.py

@@ -79,9 +79,9 @@ class TripletDataset(Dataset):
             image_files = [f for f in os.listdir(class_path) 
                           if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
             
-            # 限制每个类别的样本数
-            if len(image_files) > self.samples_per_class:
-                image_files = random.sample(image_files, self.samples_per_class)
+            # 限制每个类别的样本数,没有必要限制
+            # if len(image_files) > self.samples_per_class:
+            #     image_files = random.sample(image_files, self.samples_per_class)
             
             for img_file in image_files:
                 img_path = os.path.join(class_path, img_file)
@@ -551,7 +551,7 @@ def main():
     train_dataset = TripletDataset(
         dataset_path=settings.TRAIN_DATA_DIR,
         transform=transform_train,
-        samples_per_class=200
+        samples_per_class=550
     )
     
     val_dataset = TripletDataset(