search

kobiso / CBAM-keras

CBAM implementation on Keras
MIT License
360 stars 133 forks source link

I updated CBAM-keras Code for tf 2.0 #5

Open Lukious opened 3 years ago

Lukious commented 3 years ago

HI! , The code has been updated to support the latest package for maintainability. Now it supports not only the most recent keras versions but also tf 2.2. And I also added requirements.txt for users. (Removed some caches too)

I Tested on the following configuration under Windows 10 x64: Python 3.7 tensorflow-gpu 1.15.2 CUDA 10.0 cuDNN 7.6.4

And I also changed all '._keras_shape' to '.shape' for maintainability

Thanks.

안녕하세요 CBAM-keras를 쓰려고 하던 중에 의존성 패키지들이 너무 오래되고 버전 명시가 없어 환경구축이 힘든점이 있어서, 코드가 최신 패키지들을 지원하고 requiremets.txt를 통해 보다 쉽게 환경을 세팅 할 수 있도록 업데이트 하였습니다.

conda를 통한 가상환경에서 테스트를 완료 하였고 성공적으로 빌드 됨 또한 모두 확인하였습니다.

추가적으로 일부 캐시파일들과 비교적 사용되지 않는 _keras_sahpe의 일부를 shape로 바꾸어 사용 할 수 있도록 수정하였습니다.

혹시 체크 중 문제가 발생하는 부분이 있으시다면 알려주시면 수정하여 다시 PR드리겠습니다.

감사합니다.

Wanghd-MVP commented 3 years ago

Hi！ Bro，thanks for your update!