Skip to content

Commit

Permalink
making packages independent from img_str; standardized python package
Browse files Browse the repository at this point in the history
  • Loading branch information
marinagmoreira committed Mar 22, 2024
1 parent dc1bfa9 commit 3787701
Show file tree
Hide file tree
Showing 5 changed files with 18 additions and 9 deletions.
3 changes: 0 additions & 3 deletions .gitmodules
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
[submodule "isaac_msgs"]
path = communications/isaac_msgs
url = https://github.com/nasa/isaac_msgs.git
[submodule "parseq"]
path = anomaly/str/parseq
url = [email protected]:baudm/parseq.git
[submodule "craft"]
path = anomaly/str/craft
url = [email protected]:clovaai/CRAFT-pytorch.git
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,11 @@ libraries and build them as Python packages.

CRAFT-Pytorch

https://github.com/clovaai/CRAFT-pytorch
pip install https://github.com/marinagmoreira/CRAFT-pytorch.git#egg=craft

PARSeq

https://github.com/baudm/parseq
pip install https://github.com/marinagmoreira/parseq.git@focal#egg=parseq

The Image STR package also contains a requirements file with all the other necessary Python packages.

Expand Down
File renamed without changes.
2 changes: 1 addition & 1 deletion anomaly/image_str/scripts/image_str/parse_img.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@
from craft.craft import CRAFT
from IPython.display import HTML
from IPython.display import display as idisplay
from parseq.strhub.data.module import SceneTextDataModule
from PIL import Image
from strhub.data.module import SceneTextDataModule
from torch.autograd import Variable
from tqdm import tqdm

Expand Down
18 changes: 15 additions & 3 deletions anomaly/image_str/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,23 @@
from catkin_pkg.python_setup import generate_distutils_setup

d = generate_distutils_setup(
packages=["image_str", "craft", "parseq"],
packages=["image_str"],
package_dir={"": "scripts"},
install_requires=[
"CRAFT-pytorch @ git+ssh://[email protected]/clovaai/CRAFT-pytorch.git",
"parseq @ git+ssh://[email protected]/baudm/parseq.git",
"strhub>=4.7.1",
"craft>=4.7.1",
"gdown>=4.7.1",
"ipython>=8.12.2",
"ipywidgets>=8.0.7",
"jellyfish>=1.0.0",
"matplotlib>=3.7.2",
"numpy>=1.24.4",
"opencv_python>=4.8.0.74",
"pandas>=2.0.3",
"Pillow>=10.0.0",
"skimage>=0.0",
"torch>=2.0.1",
"tqdm>=4.65.0",
],
)

Expand Down

0 comments on commit 3787701

Please sign in to comment.