瀏覽代碼

add negative sample training doc

will-jl944 4 年之前
父節點
當前提交
97dc8593db
共有 2 個文件被更改,包括 3 次插入0 次删除
  1. 2 0
      docs/apis/datasets.md
  2. 1 0
      paddlex/cv/datasets/coco.py

+ 2 - 0
docs/apis/datasets.md

@@ -43,6 +43,7 @@ paddlex.datasets.VOCDetection(data_dir, file_list, label_list, transforms=None,
 > > * **transforms** (paddlex.transforms): 数据集中每个样本的预处理/增强算子,详见[paddlex.transforms](./transforms/transforms.md)。  
 > > * **num_workers** (int|str):数据集中样本在预处理过程中的进程数。默认为'auto'。当设为'auto'时,根据系统的实际CPU核数设置`num_workers`: 如果CPU核数的一半大于8,则`num_workers`为8,否则为CPU核数的一半。
 > > * **shuffle** (bool): 是否需要对数据集中样本打乱顺序。默认为False。
+> > * **allow_empty** (bool): 是否加载负样本。默认为False。
 
 ## <h2 id="1">paddlex.datasets.CocoDetection</h2>
 > **用于实例分割/目标检测模型**  
@@ -61,6 +62,7 @@ paddlex.datasets.CocoDetection(data_dir, ann_file, transforms=None, num_workers=
 > > * **transforms** (paddlex.transforms): 数据集中每个样本的预处理/增强算子,详见[paddlex.transforms](./transforms/transforms.md)。
 > > * **num_workers** (int|str):数据集中样本在预处理过程中的线程或进程数。默认为'auto'。当设为'auto'时,根据系统的实际CPU核数设置`num_workers`: 如果CPU核数的一半大于8,则`num_workers`为8,否则为CPU核数的一半。  
 > > * **shuffle** (bool): 是否需要对数据集中样本打乱顺序。默认为False。
+> > * **allow_empty** (bool): 是否加载负样本。默认为False。
 
 ## <h2 id="1">paddlex.datasets.SegDataset</h2>
 > **用于语义分割模型**  

+ 1 - 0
paddlex/cv/datasets/coco.py

@@ -33,6 +33,7 @@ class CocoDetection(VOCDetection):
         num_workers (int|str): 数据集中样本在预处理过程中的线程或进程数。默认为'auto'。当设为'auto'时,根据
             系统的实际CPU核数设置`num_workers`: 如果CPU核数的一半大于8,则`num_workers`为8,否则为CPU核数的一半。
         shuffle (bool): 是否需要对数据集中样本打乱顺序。默认为False。
+        allow_empty (bool): 是否加载负样本。默认为False。
     """
 
     def __init__(self,