From 264e7a160d61ca53ddefc89227e31c8cd5828285 Mon Sep 17 00:00:00 2001 From: Npsking Date: Sun, 1 Aug 2021 18:50:34 +0530 Subject: [PATCH] initial commit --- conda-cpu.yml | 15 +++++++++++++++ conda-gpu.yml | 17 +++++++++++++++++ requirements-gpu.txt | 8 ++++++++ requirements.txt | 8 ++++++++ 4 files changed, 48 insertions(+) create mode 100644 conda-cpu.yml create mode 100644 conda-gpu.yml create mode 100644 requirements-gpu.txt create mode 100644 requirements.txt diff --git a/conda-cpu.yml b/conda-cpu.yml new file mode 100644 index 0000000..396e082 --- /dev/null +++ b/conda-cpu.yml @@ -0,0 +1,15 @@ +name: yolov4-cpu + +dependencies: + - python==3.7 + - pip + - matplotlib + - opencv + - pip: + - opencv-python==4.1.1.26 + - lxml + - tqdm + - tensorflow==2.3.0rc0 + - absl-py + - easydict + - pillow \ No newline at end of file diff --git a/conda-gpu.yml b/conda-gpu.yml new file mode 100644 index 0000000..77ce711 --- /dev/null +++ b/conda-gpu.yml @@ -0,0 +1,17 @@ +name: yolov4-gpu + +dependencies: + - python==3.7 + - pip + - matplotlib + - opencv + - cudnn + - cudatoolkit==10.1.243 + - pip: + - tensorflow-gpu==2.3.0rc0 + - opencv-python==4.1.1.26 + - lxml + - tqdm + - absl-py + - easydict + - pillow diff --git a/requirements-gpu.txt b/requirements-gpu.txt new file mode 100644 index 0000000..c960a90 --- /dev/null +++ b/requirements-gpu.txt @@ -0,0 +1,8 @@ +tensorflow-gpu==2.3.0rc0 +opencv-python==4.1.1.26 +lxml +tqdm +absl-py +matplotlib +easydict +pillow diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..5c68a54 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,8 @@ +opencv-python==4.1.1.26 +lxml +tqdm +tensorflow==2.3.0rc0 +absl-py +easydict +matplotlib +pillow