Internal change

Zarjagen · tensorflower-gardener · commit 2665aad46e42 · 2023-05-15T12:54:47.000-07:00
PiperOrigin-RevId: 532196880
diff --git a/official/projects/yolo/configs/yolov7.py b/official/projects/yolo/configs/yolov7.py
@@ -332,7 +332,7 @@ def coco_yolov7_tiny() -> cfg.ExperimentConfig:
   ]
 
   config.task.model.loss.cls_weight = 0.5
-  config.task.model.loss.obj_weight = 1.0
+  config.task.model.loss.obj_weight = (416 / 640) ** 2
   config.task.train_data.parser.aug_rand_translate = 0.1
   config.task.train_data.parser.mosaic.mixup_frequency = 0.05
   config.task.train_data.parser.mosaic.aug_scale_min = 0.5

Original file line number	Diff line number	Diff line change
`@@ -332,7 +332,7 @@ def coco_yolov7_tiny() -> cfg.ExperimentConfig:`
`332`	`332`	`]`
`333`	`333`
`334`	`334`	`config.task.model.loss.cls_weight = 0.5`
`335`		`- config.task.model.loss.obj_weight = 1.0`
	`335`	`+ config.task.model.loss.obj_weight = (416 / 640) ** 2`
`336`	`336`	`config.task.train_data.parser.aug_rand_translate = 0.1`
`337`	`337`	`config.task.train_data.parser.mosaic.mixup_frequency = 0.05`
`338`	`338`	`config.task.train_data.parser.mosaic.aug_scale_min = 0.5`