Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

I updated CBAM-keras Code for tf 2.0 #5

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

Lukious
Copy link

@Lukious Lukious commented Feb 8, 2021

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
Copy link

Hi! Bro,thanks for your update!

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants