Commit 63098752 authored by Rayyyyy's avatar Rayyyyy
Browse files

Add icon and SCNet,coco128 tiny datasets

parent 7a81732e
......@@ -179,8 +179,8 @@ def build(image_set, args, eval_in_training_set):
assert root.exists(), f"provided COCO path {root} does not exist"
mode = "instances"
PATHS = {
"train": (root / "images/train2017", root / "annotations" / f"{mode}_train2017.json"),
"val": (root / "images/val2017", root / "annotations" / f"{mode}_val2017.json"),
"train": (root / "train2017", root / "annotations" / f"{mode}_train2017.json"),
"val": (root / "val2017", root / "annotations" / f"{mode}_val2017.json"),
}
img_folder, ann_file = PATHS[image_set]
......
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
RUN source /opt/dtk/env.sh
COPY requirements.txt requirements.txt
RUN pip3 install -r requirements.txt
\ No newline at end of file
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
\ No newline at end of file
icon.png

77.3 KB

Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment