This fixes the ModuleNotFoundError issues that occurred during clean installations.
Changes:
- Embedded colormnet_source dependencies directly into repository as:
- colormnet_model/ (model architecture)
- colormnet_inference/ (inference pipeline)
- colormnet_util/ (utility functions)
- colormnet_dataset/ (data transforms and utilities)
- Updated all internal imports to use new embedded paths
- Removed external colormnet_source path dependency from __init__.py
- Added graceful fallback for spatial_correlation_sampler CUDA extension
- Fixed shape calculation bugs in PyTorch fallback implementation
- Added missing dependencies (hickle, scikit-image) to auto-install
The repository is now self-contained and can be installed without requiring
the separate colormnet_source directory.
Tested and verified:
- Node imports work correctly
- Colorization pipeline runs successfully
- Model loads and processes video frames
Generated with [Claude Code](https://claude.com/claude-code)
Co-Authored-By: Claude <noreply@anthropic.com>