8000 Add dependencies needed for image classification (#291) · SerAcero/python-tutorial@c211b81 · GitHub
[go: up one dir, main page]

Skip to content

Commit c211b81

Browse files
authored
Add dependencies needed for image classification (empa-scientific-it#291)
1 parent 5b54842 commit c211b81

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

binder/environment.yml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
name: python-tutorial
33
channels:
44
- conda-forge
5+
- pytorch
56
dependencies:
67
- python=3.10
78
- pip
@@ -25,3 +26,10 @@ dependencies:
2526
- tenacity
2627
- markdown2
2728
- python-dotenv
29+
- pillow
30+
- opencv-python
31+
- torch
32+
- torchaudio
33+
- torchvision
34+
- albumentations
35+
- grad-cam

0 commit comments

Comments
 (0)
0