search

NotJoeMartinez / APDL_Quality_Control

Computer Vision Quality control for the Advanced Particle Detector Laboratory at Texas Tech University
0 stars 1 forks source link

Make input shape dynamic #21

Closed NotJoeMartinez closed 3 years ago

NotJoeMartinez commented 3 years ago

In test_model.py the input shape is not dynamic making you manually put in the shape this is time consuming.

NotJoeMartinez commented 3 years ago

Fixed in cd5807c

Modify the size argument in the main function to change the shape

test_model.py

line 25:

 model_name=MOST_RECENT_MODEL, size=(480,480)):

line 208:

data = np.ndarray(shape=(1, size[0], size[1], 3), dtype=np.float32)