Removed original yolo repo from docker image (#200)

* added /app/repo into sys.path * docker updated
supervisely-ecosystem · May 29, 2023 · 4b4d428 · 4b4d428
1 parent 98bbb98
commit 4b4d428
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/supervisely/serve/config.json b/supervisely/serve/config.json
@@ -11,7 +11,7 @@
     "serve"
   ], 
   "description": "Deploy model as REST API service",
-  "docker_image": "supervisely/yolov5:1.0.0",
+  "docker_image": "supervisely/yolov5:1.0.1",
   "instance_version": "6.6.7",
   "entrypoint": "python -m uvicorn main:m.app --app-dir ./supervisely/serve/src --host 0.0.0.0 --port 8000 --ws websockets",
   "port": 8000,

diff --git a/supervisely/train/config.json b/supervisely/train/config.json
@@ -10,7 +10,7 @@
     "train"
   ],
   "description": "Dashboard to configure and monitor training",
-  "docker_image": "supervisely/yolov5:1.0.0",
+  "docker_image": "supervisely/yolov5:1.0.1",
   "instance_version": "6.4.57",
   "main_script": "supervisely/train/src/sly_train.py",
   "gui_template": "supervisely/train/src/gui.html",

diff --git a/supervisely/train/src/sly_train_globals.py b/supervisely/train/src/sly_train_globals.py
@@ -18,6 +18,11 @@
 sys.path.append(ui_sources_dir)
 sly.logger.info(f"Added to sys.path: {ui_sources_dir}")
 
+# currently PYTHONPATH is ":/app/repo"
+python_path = os.environ.get("PYTHONPATH", "/app/repo").replace(":", "")
+sys.path.insert(0, python_path)
+
+
 load_dotenv(os.path.join(root_source_dir, "supervisely", "train", "debug.env"))
 load_dotenv(os.path.join(root_source_dir, "supervisely", "train", "secret_debug.env"), override=True)