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