This commit is contained in:
IDONTSUDO 2024-04-26 15:44:09 +03:00
parent e155b4a2a1
commit c4ddb3dc8c
23 changed files with 333 additions and 132 deletions

View file

@ -4,9 +4,13 @@ import os.path
from pathlib import Path
parser = argparse.ArgumentParser()
parser.add_argument("--path")
parser.add_argument("--name")
parser.add_argument("--path", required=True, help="Path for dataset")
parser.add_argument("--name", required=True, help="String with result weights name")
parser.add_argument("--datasetName", required=True, help="String with dataset name")
parser.add_argument("--outpath", default="weights", help="Output path for weights")
parser.add_argument("--type", default="ObjectDetection", help="Type of implementation")
parser.add_argument("--epoch", default=3, help="How many training epochs")
args = parser.parse_args()