Update gitignore

.gitignore

@@ -37,6 +37,7 @@ data/*
 !data/coco_*.txt
 !data/trainvalno5k.shapes
 !data/*.sh
+!data/watermark.yaml
 
 pycocotools/*
 results*.txt

data/watermark.yaml

@@ -0,0 +1,17 @@
+# Original watermark dataset
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+train: ./data/images/train/  # 3600 images
+val: ./data/images/test/  # 900 images
+
+# number of classes
+nc: 1
+
+# class names
+names: ['watermark']
+
+# Print classes
+# with open('data/coco.yaml') as f:
+#   d = yaml.load(f, Loader=yaml.FullLoader)  # dict
+#   for i, x in enumerate(d['names']):
+#     print(i, x)