Skip to content

Commit

Permalink
update enviornment
Browse files Browse the repository at this point in the history
  • Loading branch information
Xiaofei Huang committed Jul 5, 2024
1 parent cd21fed commit 7943e83
Show file tree
Hide file tree
Showing 11 changed files with 15,097 additions and 6,242 deletions.
369 changes: 184 additions & 185 deletions README.md

Large diffs are not rendered by default.

2 changes: 1 addition & 1 deletion experiments/coco/hrnet/w48_384x288_adam_lr1e-3_infant.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ MODEL:
INIT_WEIGHTS: true
NAME: adaptive_pose_hrnet
NUM_JOINTS: 17
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
PRETRAINED: ''#models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
TARGET_TYPE: gaussian
IMAGE_SIZE:
- 288
Expand Down
167 changes: 167 additions & 0 deletions fidip_env.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,167 @@
name: fidip
channels:
- pytorch
- nvidia
- defaults
dependencies:
- _libgcc_mutex=0.1
- _openmp_mutex=5.1
- abseil-cpp=20230802.0
- aiosignal=1.2.0
- blas=1.0
- brotli-python=1.0.9
- bzip2=1.0.8
- c-ares=1.19.1
- ca-certificates=2024.3.11
- charset-normalizer=2.0.4
- cuda-cudart=12.1.105
- cuda-cupti=12.1.105
- cuda-libraries=12.1.0
- cuda-nvrtc=12.1.105
- cuda-nvtx=12.1.105
- cuda-opencl=12.5.39
- cuda-runtime=12.1.0
- cuda-version=12.5
- expat=2.6.2
- ffmpeg=4.3
- freetype=2.12.1
- gmp=6.2.1
- gnutls=3.6.15
- google-auth-oauthlib=0.4.1
- grpc-cpp=1.48.2
- gtest=1.14.0
- intel-openmp=2023.1.0
- jpeg=9e
- lame=3.100
- lcms2=2.12
- ld_impl_linux-64=2.38
- lerc=3.0
- libcublas=12.1.0.26
- libcufft=11.0.2.4
- libcufile=1.10.0.4
- libcurand=10.3.6.39
- libcusolver=11.4.4.55
- libcusparse=12.0.2.55
- libdeflate=1.17
- libffi=3.4.4
- libgcc-ng=11.2.0
- libgomp=11.2.0
- libiconv=1.16
- libidn2=2.3.4
- libjpeg-turbo=2.0.0
- libnpp=12.0.2.50
- libnvjitlink=12.1.105
- libnvjpeg=12.1.1.14
- libpng=1.6.39
- libprotobuf=3.20.3
- libstdcxx-ng=11.2.0
- libtasn1=4.19.0
- libtiff=4.5.1
- libunistring=0.9.10
- libuuid=1.41.5
- libwebp-base=1.3.2
- llvm-openmp=14.0.6
- lz4-c=1.9.4
- mkl=2023.1.0
- mkl_fft=1.3.8
- mkl_random=1.2.4
- ncurses=6.4
- nettle=3.7.3
- numpy-base=1.26.4
- openh264=2.1.1
- openjpeg=2.4.0
- openssl=3.0.14
- pyasn1=0.4.8
- pyasn1-modules=0.2.8
- pycparser=2.21
- python=3.12.4
- pytorch=2.3.1
- pytorch-cuda=12.1
- pytorch-mutex=1.0
- re2=2022.04.01
- readline=8.2
- requests-oauthlib=1.3.0
- rsa=4.7.2
- six=1.16.0
- sqlite=3.45.3
- tbb=2021.8.0
- tensorboard=2.6.0
- tensorboard-plugin-wit=1.6.0
- tk=8.6.14
- typing_extensions=4.11.0
- xz=5.4.6
- yaml=0.2.5
- zlib=1.2.13
- zstd=1.5.5
- pip:
- absl-py==2.1.0
- aiohttp==3.9.5
- attrs==23.1.0
- blinker==1.6.2
- brotli==1.0.9
- cachetools==5.3.3
- certifi==2024.6.2
- cffi==1.16.0
- click==8.1.7
- contourpy==1.2.1
- cryptography==42.0.5
- cycler==0.12.1
- cython==3.0.10
- dsntnn==0.5.3
- easydict==1.13
- filelock==3.13.1
- fonttools==4.53.1
- frozenlist==1.4.0
- fsspec==2024.6.1
- google-auth==2.29.0
- grpcio==1.48.2
- idna==3.7
- imageio==2.34.2
- jinja2==3.1.4
- json-tricks==3.17.3
- kiwisolver==1.4.5
- lazy-loader==0.4
- markdown==3.4.1
- markupsafe==2.1.3
- matplotlib==3.9.1
- mkl-fft==1.3.8
- mkl-random==1.2.4
- mkl-service==2.4.0
- mpmath==1.3.0
- multidict==6.0.4
- networkx==3.3
- numpy==1.26.4
- oauthlib==3.2.2
- opencv-python==4.10.0.84
- packaging==24.1
- pandas==2.2.2
- pillow==10.3.0
- pip==24.0
- protobuf==3.20.3
- pycocotools==2.0.8
- pyjwt==2.8.0
- pyopenssl==24.0.0
- pyparsing==3.1.2
- pysocks==1.7.1
- python-dateutil==2.9.0.post0
- pytz==2024.1
- pyyaml==6.0.1
- requests==2.32.2
- scikit-image==0.24.0
- scipy==1.14.0
- setuptools==69.5.1
- shapely==2.0.4
- sympy==1.12
- tensorboardx==2.6.2.2
- tifffile==2024.7.2
- torch==2.3.1
- torchaudio==2.3.1
- torchvision==0.18.1
- typing-extensions==4.11.0
- tzdata==2024.1
- urllib3==2.2.2
- werkzeug==3.0.3
- wheel==0.43.0
- yacs==0.1.8
- yarl==1.9.3

12 changes: 6 additions & 6 deletions lib/dataset/syrip.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,8 +191,8 @@ def _load_syrip_keypoint_annotation_kernal(self, index):
if max(obj['keypoints']) == 0:
continue

joints_3d = np.zeros((self.num_joints, 3), dtype=np.float)
joints_3d_vis = np.zeros((self.num_joints, 3), dtype=np.float)
joints_3d = np.zeros((self.num_joints, 3), dtype=np.float32)
joints_3d_vis = np.zeros((self.num_joints, 3), dtype=np.float32)
for ipt in range(self.num_joints):
joints_3d[ipt, 0] = obj['keypoints'][ipt * 3 + 0]
joints_3d[ipt, 1] = obj['keypoints'][ipt * 3 + 1]
Expand Down Expand Up @@ -284,9 +284,9 @@ def _load_syrip_person_detection_results(self):
num_boxes = num_boxes + 1

center, scale = self._box2cs(box)
joints_3d = np.zeros((self.num_joints, 3), dtype=np.float)
joints_3d = np.zeros((self.num_joints, 3), dtype=np.float32)
joints_3d_vis = np.ones(
(self.num_joints, 3), dtype=np.float)
(self.num_joints, 3), dtype=np.float32)
kpt_db.append({
'image': img_name,
'center': center,
Expand Down Expand Up @@ -325,7 +325,7 @@ def evaluate(self, cfg, preds, output_dir, all_boxes, img_path,
_img = int(img_path[idx].split('/')[-1][5:-4])
else:
_img = int(img_path[idx][-16:-4])
print(_img)
#print(_img)
_kpts.append({
'keypoints': kpt,
'center': all_boxes[idx][0:2],
Expand Down Expand Up @@ -426,7 +426,7 @@ def _syrip_keypoint_results_one_category_kernel(self, data_pack):
_key_points = np.array([img_kpts[k]['keypoints']
for k in range(len(img_kpts))])
key_points = np.zeros(
(_key_points.shape[0], self.num_joints * 3), dtype=np.float
(_key_points.shape[0], self.num_joints * 3), dtype=np.float32
)

for ipt in range(self.num_joints):
Expand Down
Loading

0 comments on commit 7943e83

Please sign in to comment.