Skip to content

Commit 6fb12eb

Browse files
committed
slight version changes
1 parent 8974d91 commit 6fb12eb

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ Built with simplicity in mind, **ImageAI**
2828
Eventually, **ImageAI** will provide support for a wider and more specialized aspects of Computer Vision
2929

3030

31-
**New Release : ImageAI 3.0.0**
31+
**New Release : ImageAI 3.0.1**
3232

3333
What's new:
3434
- PyTorch backend

imageai/backend_check/backend_check.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,6 @@
66
import tensorflow
77
import keras
88

9-
raise RuntimeError("Dependency error!!! It appears you are trying to use ImageAI with a Tensorflow backend. ImageAI now uses PyTorch as backed as from version 3.0.0 . If you want to use the Tensorflow models or a customly trained '.h5' model, install ImageAI 2.1.6 or earlier. To use the latest Pytorch models, see the documentation in https://imageai.readthedocs.io/")
9+
raise RuntimeError("Dependency error!!! It appears you are trying to use ImageAI with a Tensorflow backend. ImageAI now uses PyTorch as backed as from version 3.0.1 . If you want to use the Tensorflow models or a customly trained '.h5' model, install ImageAI 2.1.6 or earlier. To use the latest Pytorch models, see the documentation in https://imageai.readthedocs.io/")
1010
except:
1111
raise RuntimeError("Dependency error!!! PyTorch and TorchVision are not installed. Please see installation instructions in the documentation https://imageai.readthedocs.io/")

imageai/backend_check/model_extension.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,6 @@
22

33
def extension_check(file_path: str):
44
if file_path.endswith(".h5"):
5-
raise RuntimeError("You are trying to use a Tensorflow model with ImageAI. ImageAI now uses PyTorch as backed as from version 3.0.0 . If you want to use the Tensorflow models or a customly trained '.h5' model, install ImageAI 2.1.6 or earlier. To use the latest Pytorch models, see the documentation in https://imageai.readthedocs.io/")
5+
raise RuntimeError("You are trying to use a Tensorflow model with ImageAI. ImageAI now uses PyTorch as backed as from version 3.0.1 . If you want to use the Tensorflow models or a customly trained '.h5' model, install ImageAI 2.1.6 or earlier. To use the latest Pytorch models, see the documentation in https://imageai.readthedocs.io/")
66
elif file_path.endswith(".pt") == False and file_path.endswith(".pth") == False:
77
raise ValueError(f"Invalid model file {os.path.basename(file_path)}. Please parse in a '.pt' and '.pth' model file.")

0 commit comments

Comments
 (0)