Skip to content
New issue

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

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

Already on GitHub? Sign in 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!

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

Successfully merging this pull request may close these issues.

2 participants