- Integrate Convolutional Block Attention Module (CBAM) for improved feature focus - Implement temperature scaling and confidence-based status reporting - Automate dataset acquisition using kagglehub - Update ONNX opset to 18 and refine preprocessing validation
14 lines
339 B
Plaintext
14 lines
339 B
Plaintext
tensorflow==2.19.0 # CPU + Colab; for local GPU, install tensorflow[and-cuda]
|
|
tensorflowjs==4.22.0
|
|
gdown # download dataset from Google Drive
|
|
kagglehub # download dataset from Kaggle
|
|
numpy
|
|
matplotlib
|
|
seaborn
|
|
pillow
|
|
split-folders
|
|
scikit-learn
|
|
tf2onnx
|
|
onnxruntime
|
|
huggingface_hub
|