Tensorflow2 migration - #47
Open
mustafa-meco wants to merge 6 commits into
Open
Conversation
- Replace Mask_RCNN (TF1) with Mask_RCNN_TF2 for GPU compatibility - Update inference.py with TF2 optimizations and GPU memory growth - Update requirements.txt to use TensorFlow >= 2.4.0 - Update Dockerfile to use TF 2.14.0 base image - Update all configuration paths and imports - Remove old Mask_RCNN directory and dependencies - Update .gitignore and .dockerignore for new structure - Exclude virtual environments and temporary files from repository 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
- Upgrade base image from tensorflow:2.14.0-gpu to tensorflow:latest-gpu - Update all Python package versions to latest compatible versions - Modernize pip installation practices (use python3 -m pip) - Replace deprecated libpython2.7-dev with python3-dev - Update package installation methods (setup.py → pip install -e .) - Add modern build tools and improve error handling - Ensure compatibility with Python 3.11+ and latest TensorFlow Updated package versions: - numpy: 1.18.1 → >=1.24.0,<2.0 - opencv-python: 4.2.0.34 → >=4.8.0 - tensorflow: >=2.4.0 → >=2.14.0 - scikit-image: 0.16.2 → >=0.21.0 - cython: 0.29.21 → >=3.0.0 - And many other packages updated to latest versions 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
- Add docker-compose.yml with GPU support and volume mounts - Add .env.example template for environment configuration - Add comprehensive DOCKER_USAGE.md with step-by-step instructions - Add config.yml.example template with all processing options - Add setup.sh script for automated environment setup - Add download-models.sh script for fetching pre-trained weights Features: - Full GPU support with NVIDIA Container Toolkit - Flexible volume mounts for data, models, and outputs - Development mode with Jupyter Lab - Health checks and monitoring - Comprehensive troubleshooting guide - Example configurations for IR and RGB processing Ready for production deployment with simple docker-compose up\! 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
…b management - Added Pydantic models for API requests and responses in `api/models.py`. - Created processing logic in `api/processors.py` to handle various PV-Hawk tasks. - Developed FastAPI server in `api/server.py` to expose REST endpoints for job management and processing. - Introduced job management functionality to track and control processing jobs. - Added system and GPU information endpoints for monitoring. - Included requirements for FastAPI and dependencies in `api/requirements.txt`. - Updated `docker-compose.yml` to support API mode with GPU capabilities. - Created startup script `start_api.py` for launching the API server with configurable options.
- Removed REST API implementation (moved to PV-Hawk-API private repo) - Removed API-specific testing infrastructure - Cleaned up tests documentation - Kept public TensorFlow 2 migration and core improvements - Maintained compatibility with upstream project
Author
|
add tensorflow 2 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
No description provided.