Commit 6853cd3

committed

addd new feature

1 parent fda4a40 commit 6853cd3Copy full SHA for 6853cd3

File tree

+755

-0

lines changed

+755

-0

lines changed

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,8 @@ def get_args():`
`28`	`28`	`parser.add_argument('--use_nano', action='store_true')`
`29`	`29`	`parser.add_argument('--use_vit', action='store_true')`
`30`	`30`
	`31`	`+ parser.add_argument('--use_yolo', action='store_true', help='Use YOLO for object detection')`
	`32`	`+`
`31`	`33`	`args = parser.parse_args()`
`32`	`34`
`33`	`35`	`return args`

Comments

(0)